.gitignore
CHANGELOG.md
CODEOWNERS
CODE_OF_CONDUCT.md
Dockerfile
LICENSE.txt
README.md
SECURITY.md
requirements.txt
setup.py
docs/.nojekyll
docs/Makefile
docs/example_envs.html
docs/example_envs.tag_continuous.html
docs/example_envs.tag_gridworld.html
docs/genindex.html
docs/index.html
docs/make.bat
docs/modules.html
docs/objects.inv
docs/py-modindex.html
docs/search.html
docs/searchindex.js
docs/warp_drive.html
docs/warp_drive.managers.html
docs/warp_drive.training.algorithms.html
docs/warp_drive.training.html
docs/warp_drive.training.models.html
docs/warp_drive.training.utils.html
docs/warp_drive.utils.html
docs/_sources/data_manager.rst.txt
docs/_sources/example_envs.rst.txt
docs/_sources/example_envs.tag_continuous.rst.txt
docs/_sources/example_envs.tag_gridworld.rst.txt
docs/_sources/function_manager.rst.txt
docs/_sources/index.rst.txt
docs/_sources/modules.rst.txt
docs/_sources/warp_drive.managers.rst.txt
docs/_sources/warp_drive.rst.txt
docs/_sources/warp_drive.training.algorithms.rst.txt
docs/_sources/warp_drive.training.models.rst.txt
docs/_sources/warp_drive.training.rst.txt
docs/_sources/warp_drive.training.utils.rst.txt
docs/_sources/warp_drive.utils.rst.txt
docs/_static/basic.css
docs/_static/doctools.js
docs/_static/documentation_options.js
docs/_static/file.png
docs/_static/jquery-3.5.1.js
docs/_static/jquery.js
docs/_static/language_data.js
docs/_static/minus.png
docs/_static/plus.png
docs/_static/pygments.css
docs/_static/searchtools.js
docs/_static/underscore-1.13.1.js
docs/_static/underscore.js
docs/_static/css/badge_only.css
docs/_static/css/theme.css
docs/_static/css/fonts/Roboto-Slab-Bold.woff
docs/_static/css/fonts/Roboto-Slab-Bold.woff2
docs/_static/css/fonts/Roboto-Slab-Regular.woff
docs/_static/css/fonts/Roboto-Slab-Regular.woff2
docs/_static/css/fonts/fontawesome-webfont.eot
docs/_static/css/fonts/fontawesome-webfont.svg
docs/_static/css/fonts/fontawesome-webfont.ttf
docs/_static/css/fonts/fontawesome-webfont.woff
docs/_static/css/fonts/fontawesome-webfont.woff2
docs/_static/css/fonts/lato-bold-italic.woff
docs/_static/css/fonts/lato-bold-italic.woff2
docs/_static/css/fonts/lato-bold.woff
docs/_static/css/fonts/lato-bold.woff2
docs/_static/css/fonts/lato-normal-italic.woff
docs/_static/css/fonts/lato-normal-italic.woff2
docs/_static/css/fonts/lato-normal.woff
docs/_static/css/fonts/lato-normal.woff2
docs/_static/fonts/Inconsolata-Bold.ttf
docs/_static/fonts/Inconsolata-Regular.ttf
docs/_static/fonts/Inconsolata.ttf
docs/_static/fonts/Lato-Bold.ttf
docs/_static/fonts/Lato-Regular.ttf
docs/_static/fonts/RobotoSlab-Bold.ttf
docs/_static/fonts/RobotoSlab-Regular.ttf
docs/_static/fonts/fontawesome-webfont.eot
docs/_static/fonts/fontawesome-webfont.svg
docs/_static/fonts/fontawesome-webfont.ttf
docs/_static/fonts/fontawesome-webfont.woff
docs/_static/fonts/fontawesome-webfont.woff2
docs/_static/fonts/Lato/lato-bold.eot
docs/_static/fonts/Lato/lato-bold.ttf
docs/_static/fonts/Lato/lato-bold.woff
docs/_static/fonts/Lato/lato-bold.woff2
docs/_static/fonts/Lato/lato-bolditalic.eot
docs/_static/fonts/Lato/lato-bolditalic.ttf
docs/_static/fonts/Lato/lato-bolditalic.woff
docs/_static/fonts/Lato/lato-bolditalic.woff2
docs/_static/fonts/Lato/lato-italic.eot
docs/_static/fonts/Lato/lato-italic.ttf
docs/_static/fonts/Lato/lato-italic.woff
docs/_static/fonts/Lato/lato-italic.woff2
docs/_static/fonts/Lato/lato-regular.eot
docs/_static/fonts/Lato/lato-regular.ttf
docs/_static/fonts/Lato/lato-regular.woff
docs/_static/fonts/Lato/lato-regular.woff2
docs/_static/fonts/RobotoSlab/roboto-slab-v7-bold.eot
docs/_static/fonts/RobotoSlab/roboto-slab-v7-bold.ttf
docs/_static/fonts/RobotoSlab/roboto-slab-v7-bold.woff
docs/_static/fonts/RobotoSlab/roboto-slab-v7-bold.woff2
docs/_static/fonts/RobotoSlab/roboto-slab-v7-regular.eot
docs/_static/fonts/RobotoSlab/roboto-slab-v7-regular.ttf
docs/_static/fonts/RobotoSlab/roboto-slab-v7-regular.woff
docs/_static/fonts/RobotoSlab/roboto-slab-v7-regular.woff2
docs/_static/js/badge_only.js
docs/_static/js/html5shiv-printshiv.min.js
docs/_static/js/html5shiv.min.js
docs/_static/js/modernizr.min.js
docs/_static/js/theme.js
docs/json/genindex.fjson
docs/json/index.fjson
docs/json/readthedocs-sphinx-domain-names.json
docs/json/search.fjson
docs/source/conf.py
docs/source/data_manager.rst
docs/source/example_envs.rst
docs/source/example_envs.tag_continuous.rst
docs/source/example_envs.tag_gridworld.rst
docs/source/function_manager.rst
docs/source/index.rst
docs/source/modules.rst
docs/source/warp_drive.managers.rst
docs/source/warp_drive.rst
docs/source/warp_drive.training.algorithms.rst
docs/source/warp_drive.training.models.rst
docs/source/warp_drive.training.rst
docs/source/warp_drive.training.utils.rst
docs/source/warp_drive.utils.rst
example_envs/__init__.py
example_envs/dummy_env/__init__.py
example_envs/dummy_env/test_step.cu
example_envs/dummy_env/test_step_numba.py
example_envs/dummy_env/tutorial_basics.py
example_envs/tag_continuous/__init__.py
example_envs/tag_continuous/generate_rollout_animation.py
example_envs/tag_continuous/tag_continuous.py
example_envs/tag_continuous/tag_continuous_step_numba.py
example_envs/tag_continuous/tag_continuous_step_pycuda.cu
example_envs/tag_gridworld/__init__.py
example_envs/tag_gridworld/tag_gridworld.py
example_envs/tag_gridworld/tag_gridworld_step.cu
rl_warp_drive.egg-info/PKG-INFO
rl_warp_drive.egg-info/SOURCES.txt
rl_warp_drive.egg-info/dependency_links.txt
rl_warp_drive.egg-info/requires.txt
rl_warp_drive.egg-info/top_level.txt
tests/__init__.py
tests/example_envs/__init__.py
tests/example_envs/numba_tests/test_tag_continuous.py
tests/example_envs/pycuda_tests/__init__.py
tests/example_envs/pycuda_tests/test_tag_continuous.py
tests/example_envs/pycuda_tests/test_tag_gridworld.py
tests/example_envs/pycuda_tests/test_tag_gridworld_step_cuda.py
tests/example_envs/pycuda_tests/test_tag_gridworld_step_python.py
tests/multiblocks_per_env/__init__.py
tests/multiblocks_per_env/example_envs/__init__.py
tests/multiblocks_per_env/example_envs/pycuda_tests/__init__.py
tests/multiblocks_per_env/example_envs/pycuda_tests/test_tag_continuous_multiblocks.py
tests/multiblocks_per_env/warp_drive/__init__.py
tests/multiblocks_per_env/warp_drive/pycuda_tests/__init__.py
tests/multiblocks_per_env/warp_drive/pycuda_tests/test_action_sampler_multiblocks.py
tests/multiblocks_per_env/warp_drive/pycuda_tests/test_env_reset_multiblocks.py
tests/multiblocks_per_env/warp_drive/pycuda_tests/test_function_manager_multiblocks.py
tests/warp_drive/__init__.py
tests/warp_drive/numba_tests/__init__.py
tests/warp_drive/numba_tests/test_action_sampler.py
tests/warp_drive/numba_tests/test_data_manager.py
tests/warp_drive/numba_tests/test_env_reset.py
tests/warp_drive/numba_tests/test_function_manager.py
tests/warp_drive/pycuda_tests/__init__.py
tests/warp_drive/pycuda_tests/test_action_sampler.py
tests/warp_drive/pycuda_tests/test_data_manager.py
tests/warp_drive/pycuda_tests/test_env_reset.py
tests/warp_drive/pycuda_tests/test_function_manager.py
tests/wd_training/__init__.py
tests/wd_training/pycuda_tests/__init__.py
tests/wd_training/pycuda_tests/test_env_training.py
tutorials/simple-end-to-end-example.ipynb
tutorials/tutorial-1.a-warp_drive_basics.ipynb
tutorials/tutorial-1.b-warp_drive_basics.ipynb
tutorials/tutorial-2.a-warp_drive_sampler.ipynb
tutorials/tutorial-2.b-warp_drive_sampler.ipynb
tutorials/tutorial-3-warp_drive_reset_and_log.ipynb
tutorials/tutorial-4.a-create_custom_environments_pycuda.md
tutorials/tutorial-4.b-create_custom_environments_numba.md
tutorials/tutorial-5-training_with_warp_drive.ipynb
tutorials/tutorial-6-scaling_up_training_with_warp_drive.md
tutorials/tutorial-7-training_with_warp_drive_and_pytorch_lightning.ipynb
tutorials/tutorial-7-training_with_warp_drive_and_pytorch_lightning.py
tutorials/assets/gpu_memory_model.png
tutorials/assets/horizontal_scaling.png
tutorials/assets/maximizing_GPU_usage.gif
tutorials/assets/warpdrive_framework_overview.png
tutorials/assets/tag_continuous_training/run_config.yaml
tutorials/assets/tag_continuous_training/runner_1000010000.state_dict
tutorials/assets/tag_continuous_training/t=0.png
tutorials/assets/tag_continuous_training/t=250.png
tutorials/assets/tag_continuous_training/t=500.png
tutorials/assets/tag_continuous_training/tagger_1000010000.state_dict
warp_drive/__init__.py
warp_drive/env_cpu_gpu_consistency_checker.py
warp_drive/env_wrapper.py
warp_drive/cuda_bin/test_build.fatbin
warp_drive/cuda_includes/Makefile
warp_drive/cuda_includes/core_service.h
warp_drive/cuda_includes/template_env_config.h
warp_drive/cuda_includes/template_env_runner.cu
warp_drive/cuda_includes/test_build.cu
warp_drive/cuda_includes/test_build_multiblocks.cu
warp_drive/cuda_includes/core/array_indexing_util.cu
warp_drive/cuda_includes/core/array_indexing_util.h
warp_drive/cuda_includes/core/env_dim_mapper.h
warp_drive/cuda_includes/core/env_thread_sync.cu
warp_drive/cuda_includes/core/env_thread_sync.h
warp_drive/cuda_includes/core/log.cu
warp_drive/cuda_includes/core/log.h
warp_drive/cuda_includes/core/rand_reset.h
warp_drive/cuda_includes/core/random.cu
warp_drive/cuda_includes/core/random.h
warp_drive/cuda_includes/core/reset.cu
warp_drive/cuda_includes/core/reset.h
warp_drive/managers/__init__.py
warp_drive/managers/data_manager.py
warp_drive/managers/function_manager.py
warp_drive/managers/numba_managers/__init__.py
warp_drive/managers/numba_managers/numba_data_manager.py
warp_drive/managers/numba_managers/numba_function_manager.py
warp_drive/managers/pycuda_managers/__init__.py
warp_drive/managers/pycuda_managers/pycuda_data_manager.py
warp_drive/managers/pycuda_managers/pycuda_function_manager.py
warp_drive/numba_includes/__init__.py
warp_drive/numba_includes/template_env_config.txt
warp_drive/numba_includes/template_env_runner.txt
warp_drive/numba_includes/test_build.py
warp_drive/numba_includes/core/__init__.py
warp_drive/numba_includes/core/log.py
warp_drive/numba_includes/core/random.py
warp_drive/numba_includes/core/reset.py
warp_drive/training/__init__.py
warp_drive/training/example_training_script_numba.py
warp_drive/training/example_training_script_pycuda.py
warp_drive/training/pytorch_lightning.py
warp_drive/training/trainer.py
warp_drive/training/algorithms/__init__.py
warp_drive/training/algorithms/a2c.py
warp_drive/training/algorithms/ppo.py
warp_drive/training/models/__init__.py
warp_drive/training/models/fully_connected.py
warp_drive/training/run_configs/default_configs.yaml
warp_drive/training/run_configs/tag_continuous.yaml
warp_drive/training/run_configs/tag_gridworld.yaml
warp_drive/training/utils/__init__.py
warp_drive/training/utils/child_process_base.py
warp_drive/training/utils/data_loader.py
warp_drive/training/utils/param_scheduler.py
warp_drive/training/utils/process_group_torch.py
warp_drive/training/utils/vertical_scaler.py
warp_drive/training/utils/device_child_process/__init__.py
warp_drive/training/utils/device_child_process/child_process_numba.py
warp_drive/training/utils/device_child_process/child_process_pycuda.py
warp_drive/training/utils/distributed_train/distributed_trainer_numba.py
warp_drive/training/utils/distributed_train/distributed_trainer_pycuda.py
warp_drive/training/utils/single_device_context/__init__.py
warp_drive/training/utils/single_device_context/device_context_numba.py
warp_drive/training/utils/single_device_context/device_context_pycuda.py
warp_drive/utils/__init__.py
warp_drive/utils/architecture_validate.py
warp_drive/utils/argument_fix.py
warp_drive/utils/common.py
warp_drive/utils/constants.py
warp_drive/utils/data_feed.py
warp_drive/utils/env_registrar.py
warp_drive/utils/gpu_environment_context.py
warp_drive/utils/recursive_obs_dict_to_spaces_dict.py
warp_drive/utils/numba_utils/__init__.py
warp_drive/utils/numba_utils/misc.py
warp_drive/utils/pycuda_utils/__init__.py
warp_drive/utils/pycuda_utils/misc.py