├── .clang-format ├── .github ├── scripts │ └── run_ci_tests.sh └── workflows │ ├── build.yml │ ├── docs.yaml │ └── format.yaml ├── CMakeLists.txt ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── SECURITY.md ├── docker ├── Dockerfile ├── Dockerfile_gnu └── Dockerfile_gnu_cpuonly ├── docs ├── Doxyfile ├── Makefile ├── _static │ └── style.css ├── _templates │ ├── footer.html │ └── layout.html ├── api.rst ├── api │ ├── c_api.rst │ ├── config.rst │ └── f_api.rst ├── conf.py ├── extras.rst ├── index.rst ├── installation.rst ├── requirements.txt └── usage.rst ├── examples ├── cpp │ └── cart_pole │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── config.yaml │ │ ├── config_sim.yaml │ │ ├── env.cpp │ │ ├── env.h │ │ ├── media │ │ ├── cartpole.gif │ │ └── cartpole.mp4 │ │ ├── py_env.cpp │ │ ├── python │ │ ├── initialize_models.py │ │ ├── models.py │ │ └── visualize.py │ │ └── train.cpp └── fortran │ ├── graph │ ├── CMakeLists.txt │ ├── README.md │ ├── config.yaml │ ├── connectivity.txt │ ├── generate_loss.py │ ├── generate_model.py │ ├── media │ │ └── validation_results.gif │ ├── nodes.txt │ ├── train.f90 │ └── visualize.py │ └── simulation │ ├── CMakeLists.txt │ ├── README.md │ ├── config_fcn_torchscript.yaml │ ├── config_mlp_native.yaml │ ├── generate_fcn_model.py │ ├── media │ └── validation_results.gif │ ├── simulation.f90 │ ├── train.f90 │ ├── train_distributed.f90 │ └── visualize.py ├── requirements.txt ├── src ├── csrc │ ├── distributed.cpp │ ├── include │ │ ├── internal │ │ │ ├── base_loss.h │ │ │ ├── base_lr_scheduler.h │ │ │ ├── base_model.h │ │ │ ├── defines.h │ │ │ ├── distributed.h │ │ │ ├── exceptions.h │ │ │ ├── logging.h │ │ │ ├── losses.h │ │ │ ├── lr_schedulers.h │ │ │ ├── model_pack.h │ │ │ ├── model_state.h │ │ │ ├── model_wrapper.h │ │ │ ├── models.h │ │ │ ├── nvtx.h │ │ │ ├── param_map.h │ │ │ ├── rl │ │ │ │ ├── distributions.h │ │ │ │ ├── noise_actor.h │ │ │ │ ├── off_policy.h │ │ │ │ ├── off_policy │ │ │ │ │ ├── ddpg.h │ │ │ │ │ ├── sac.h │ │ │ │ │ └── td3.h │ │ │ │ ├── on_policy.h │ │ │ │ ├── on_policy │ │ │ │ │ └── ppo.h │ │ │ │ ├── policy.h │ │ │ │ ├── replay_buffer.h │ │ │ │ ├── rl.h │ │ │ │ ├── rollout_buffer.h │ │ │ │ └── utils.h │ │ │ ├── setup.h │ │ │ ├── tensor_list.h │ │ │ ├── training.h │ │ │ └── utils.h │ │ ├── torchfort.h │ │ ├── torchfort_config.h.in │ │ ├── torchfort_enums.h │ │ └── torchfort_rl.h │ ├── logging.cpp │ ├── losses │ │ ├── l1_loss.cpp │ │ ├── mse_loss.cpp │ │ └── torchscript_loss.cpp │ ├── lr_schedulers │ │ ├── cosine_annealing_lr.cpp │ │ ├── linear_lr.cpp │ │ ├── multistep_lr.cpp │ │ ├── polynomial_lr.cpp │ │ ├── scheduler_setup.cpp │ │ └── step_lr.cpp │ ├── model_pack.cpp │ ├── model_state.cpp │ ├── model_wrapper.cpp │ ├── models │ │ ├── mlp_model.cpp │ │ └── rl │ │ │ ├── actor_critic_model.cpp │ │ │ ├── common_models.cpp │ │ │ └── sac_model.cpp │ ├── param_map.cpp │ ├── rl │ │ ├── off_policy │ │ │ ├── ddpg.cpp │ │ │ ├── interface.cpp │ │ │ ├── sac.cpp │ │ │ └── td3.cpp │ │ ├── on_policy │ │ │ ├── interface.cpp │ │ │ └── ppo.cpp │ │ ├── policy.cpp │ │ └── utils.cpp │ ├── setup.cpp │ ├── torchfort.cpp │ ├── training.cpp │ └── utils.cpp ├── fsrc │ └── torchfort_m.F90 └── python │ └── wandb_helper.py └── tests ├── general ├── CMakeLists.txt ├── configs │ ├── l1.yaml │ ├── l1_multiarg.yaml │ ├── mse.yaml │ ├── mse_multiarg.yaml │ ├── torchscript.yaml │ ├── torchscript_multiarg.yaml │ ├── torchscript_multiarg_extra.yaml │ └── torchscript_multiout.yaml ├── scripts │ └── setup_tests.py └── test_losses.cpp ├── rl ├── CMakeLists.txt ├── configs │ ├── ddpg.yaml │ ├── ppo.yaml │ ├── sac.yaml │ └── td3.yaml ├── environments.h ├── test_distributions.cpp ├── test_interface.cpp ├── test_off_policy.cpp ├── test_on_policy.cpp ├── test_replay_buffer.cpp └── test_rollout_buffer.cpp ├── supervised ├── CMakeLists.txt ├── configs │ ├── missing_loss.yaml │ ├── missing_opt.yaml │ ├── mlp.yaml │ ├── mlp2.yaml │ ├── mlp2_gradacc.yaml │ ├── mlp3.yaml │ ├── torchscript.yaml │ ├── torchscript_multiarg.yaml │ └── torchscript_multiarg_extra.yaml ├── scripts │ └── setup_tests.py ├── test_checkpoint.cpp ├── test_distributed_training.cpp └── test_training.cpp └── test_utils.h /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/.clang-format -------------------------------------------------------------------------------- /.github/scripts/run_ci_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/.github/scripts/run_ci_tests.sh -------------------------------------------------------------------------------- /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/.github/workflows/docs.yaml -------------------------------------------------------------------------------- /.github/workflows/format.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/.github/workflows/format.yaml -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/SECURITY.md -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/Dockerfile_gnu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docker/Dockerfile_gnu -------------------------------------------------------------------------------- /docker/Dockerfile_gnu_cpuonly: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docker/Dockerfile_gnu_cpuonly -------------------------------------------------------------------------------- /docs/Doxyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/Doxyfile -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/style.css: -------------------------------------------------------------------------------- 1 | .wy-nav-content { 2 | max-width: 1200px !important; 3 | } 4 | -------------------------------------------------------------------------------- /docs/_templates/footer.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/_templates/footer.html -------------------------------------------------------------------------------- /docs/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/_templates/layout.html -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/api/c_api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/api/c_api.rst -------------------------------------------------------------------------------- /docs/api/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/api/config.rst -------------------------------------------------------------------------------- /docs/api/f_api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/api/f_api.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/extras.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/extras.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/installation.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/usage.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/docs/usage.rst -------------------------------------------------------------------------------- /examples/cpp/cart_pole/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/CMakeLists.txt -------------------------------------------------------------------------------- /examples/cpp/cart_pole/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/README.md -------------------------------------------------------------------------------- /examples/cpp/cart_pole/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/config.yaml -------------------------------------------------------------------------------- /examples/cpp/cart_pole/config_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/config_sim.yaml -------------------------------------------------------------------------------- /examples/cpp/cart_pole/env.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/env.cpp -------------------------------------------------------------------------------- /examples/cpp/cart_pole/env.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/env.h -------------------------------------------------------------------------------- /examples/cpp/cart_pole/media/cartpole.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/media/cartpole.gif -------------------------------------------------------------------------------- /examples/cpp/cart_pole/media/cartpole.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/media/cartpole.mp4 -------------------------------------------------------------------------------- /examples/cpp/cart_pole/py_env.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/py_env.cpp -------------------------------------------------------------------------------- /examples/cpp/cart_pole/python/initialize_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/python/initialize_models.py -------------------------------------------------------------------------------- /examples/cpp/cart_pole/python/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/python/models.py -------------------------------------------------------------------------------- /examples/cpp/cart_pole/python/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/python/visualize.py -------------------------------------------------------------------------------- /examples/cpp/cart_pole/train.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/cpp/cart_pole/train.cpp -------------------------------------------------------------------------------- /examples/fortran/graph/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/CMakeLists.txt -------------------------------------------------------------------------------- /examples/fortran/graph/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/README.md -------------------------------------------------------------------------------- /examples/fortran/graph/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/config.yaml -------------------------------------------------------------------------------- /examples/fortran/graph/connectivity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/connectivity.txt -------------------------------------------------------------------------------- /examples/fortran/graph/generate_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/generate_loss.py -------------------------------------------------------------------------------- /examples/fortran/graph/generate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/generate_model.py -------------------------------------------------------------------------------- /examples/fortran/graph/media/validation_results.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/media/validation_results.gif -------------------------------------------------------------------------------- /examples/fortran/graph/nodes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/nodes.txt -------------------------------------------------------------------------------- /examples/fortran/graph/train.f90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/train.f90 -------------------------------------------------------------------------------- /examples/fortran/graph/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/graph/visualize.py -------------------------------------------------------------------------------- /examples/fortran/simulation/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/CMakeLists.txt -------------------------------------------------------------------------------- /examples/fortran/simulation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/README.md -------------------------------------------------------------------------------- /examples/fortran/simulation/config_fcn_torchscript.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/config_fcn_torchscript.yaml -------------------------------------------------------------------------------- /examples/fortran/simulation/config_mlp_native.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/config_mlp_native.yaml -------------------------------------------------------------------------------- /examples/fortran/simulation/generate_fcn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/generate_fcn_model.py -------------------------------------------------------------------------------- /examples/fortran/simulation/media/validation_results.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/media/validation_results.gif -------------------------------------------------------------------------------- /examples/fortran/simulation/simulation.f90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/simulation.f90 -------------------------------------------------------------------------------- /examples/fortran/simulation/train.f90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/train.f90 -------------------------------------------------------------------------------- /examples/fortran/simulation/train_distributed.f90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/train_distributed.f90 -------------------------------------------------------------------------------- /examples/fortran/simulation/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/examples/fortran/simulation/visualize.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/csrc/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/distributed.cpp -------------------------------------------------------------------------------- /src/csrc/include/internal/base_loss.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/base_loss.h -------------------------------------------------------------------------------- /src/csrc/include/internal/base_lr_scheduler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/base_lr_scheduler.h -------------------------------------------------------------------------------- /src/csrc/include/internal/base_model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/base_model.h -------------------------------------------------------------------------------- /src/csrc/include/internal/defines.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/defines.h -------------------------------------------------------------------------------- /src/csrc/include/internal/distributed.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/distributed.h -------------------------------------------------------------------------------- /src/csrc/include/internal/exceptions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/exceptions.h -------------------------------------------------------------------------------- /src/csrc/include/internal/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/logging.h -------------------------------------------------------------------------------- /src/csrc/include/internal/losses.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/losses.h -------------------------------------------------------------------------------- /src/csrc/include/internal/lr_schedulers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/lr_schedulers.h -------------------------------------------------------------------------------- /src/csrc/include/internal/model_pack.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/model_pack.h -------------------------------------------------------------------------------- /src/csrc/include/internal/model_state.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/model_state.h -------------------------------------------------------------------------------- /src/csrc/include/internal/model_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/model_wrapper.h -------------------------------------------------------------------------------- /src/csrc/include/internal/models.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/models.h -------------------------------------------------------------------------------- /src/csrc/include/internal/nvtx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/nvtx.h -------------------------------------------------------------------------------- /src/csrc/include/internal/param_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/param_map.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/distributions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/distributions.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/noise_actor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/noise_actor.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/off_policy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/off_policy.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/off_policy/ddpg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/off_policy/ddpg.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/off_policy/sac.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/off_policy/sac.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/off_policy/td3.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/off_policy/td3.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/on_policy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/on_policy.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/on_policy/ppo.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/on_policy/ppo.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/policy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/policy.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/replay_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/replay_buffer.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/rl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/rl.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/rollout_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/rollout_buffer.h -------------------------------------------------------------------------------- /src/csrc/include/internal/rl/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/rl/utils.h -------------------------------------------------------------------------------- /src/csrc/include/internal/setup.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/setup.h -------------------------------------------------------------------------------- /src/csrc/include/internal/tensor_list.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/tensor_list.h -------------------------------------------------------------------------------- /src/csrc/include/internal/training.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/training.h -------------------------------------------------------------------------------- /src/csrc/include/internal/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/internal/utils.h -------------------------------------------------------------------------------- /src/csrc/include/torchfort.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/torchfort.h -------------------------------------------------------------------------------- /src/csrc/include/torchfort_config.h.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/torchfort_config.h.in -------------------------------------------------------------------------------- /src/csrc/include/torchfort_enums.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/torchfort_enums.h -------------------------------------------------------------------------------- /src/csrc/include/torchfort_rl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/include/torchfort_rl.h -------------------------------------------------------------------------------- /src/csrc/logging.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/logging.cpp -------------------------------------------------------------------------------- /src/csrc/losses/l1_loss.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/losses/l1_loss.cpp -------------------------------------------------------------------------------- /src/csrc/losses/mse_loss.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/losses/mse_loss.cpp -------------------------------------------------------------------------------- /src/csrc/losses/torchscript_loss.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/losses/torchscript_loss.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/cosine_annealing_lr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/cosine_annealing_lr.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/linear_lr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/linear_lr.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/multistep_lr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/multistep_lr.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/polynomial_lr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/polynomial_lr.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/scheduler_setup.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/scheduler_setup.cpp -------------------------------------------------------------------------------- /src/csrc/lr_schedulers/step_lr.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/lr_schedulers/step_lr.cpp -------------------------------------------------------------------------------- /src/csrc/model_pack.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/model_pack.cpp -------------------------------------------------------------------------------- /src/csrc/model_state.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/model_state.cpp -------------------------------------------------------------------------------- /src/csrc/model_wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/model_wrapper.cpp -------------------------------------------------------------------------------- /src/csrc/models/mlp_model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/models/mlp_model.cpp -------------------------------------------------------------------------------- /src/csrc/models/rl/actor_critic_model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/models/rl/actor_critic_model.cpp -------------------------------------------------------------------------------- /src/csrc/models/rl/common_models.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/models/rl/common_models.cpp -------------------------------------------------------------------------------- /src/csrc/models/rl/sac_model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/models/rl/sac_model.cpp -------------------------------------------------------------------------------- /src/csrc/param_map.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/param_map.cpp -------------------------------------------------------------------------------- /src/csrc/rl/off_policy/ddpg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/off_policy/ddpg.cpp -------------------------------------------------------------------------------- /src/csrc/rl/off_policy/interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/off_policy/interface.cpp -------------------------------------------------------------------------------- /src/csrc/rl/off_policy/sac.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/off_policy/sac.cpp -------------------------------------------------------------------------------- /src/csrc/rl/off_policy/td3.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/off_policy/td3.cpp -------------------------------------------------------------------------------- /src/csrc/rl/on_policy/interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/on_policy/interface.cpp -------------------------------------------------------------------------------- /src/csrc/rl/on_policy/ppo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/on_policy/ppo.cpp -------------------------------------------------------------------------------- /src/csrc/rl/policy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/policy.cpp -------------------------------------------------------------------------------- /src/csrc/rl/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/rl/utils.cpp -------------------------------------------------------------------------------- /src/csrc/setup.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/setup.cpp -------------------------------------------------------------------------------- /src/csrc/torchfort.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/torchfort.cpp -------------------------------------------------------------------------------- /src/csrc/training.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/training.cpp -------------------------------------------------------------------------------- /src/csrc/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/csrc/utils.cpp -------------------------------------------------------------------------------- /src/fsrc/torchfort_m.F90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/fsrc/torchfort_m.F90 -------------------------------------------------------------------------------- /src/python/wandb_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/src/python/wandb_helper.py -------------------------------------------------------------------------------- /tests/general/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/CMakeLists.txt -------------------------------------------------------------------------------- /tests/general/configs/l1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/l1.yaml -------------------------------------------------------------------------------- /tests/general/configs/l1_multiarg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/l1_multiarg.yaml -------------------------------------------------------------------------------- /tests/general/configs/mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/mse.yaml -------------------------------------------------------------------------------- /tests/general/configs/mse_multiarg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/mse_multiarg.yaml -------------------------------------------------------------------------------- /tests/general/configs/torchscript.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/torchscript.yaml -------------------------------------------------------------------------------- /tests/general/configs/torchscript_multiarg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/torchscript_multiarg.yaml -------------------------------------------------------------------------------- /tests/general/configs/torchscript_multiarg_extra.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/torchscript_multiarg_extra.yaml -------------------------------------------------------------------------------- /tests/general/configs/torchscript_multiout.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/configs/torchscript_multiout.yaml -------------------------------------------------------------------------------- /tests/general/scripts/setup_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/scripts/setup_tests.py -------------------------------------------------------------------------------- /tests/general/test_losses.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/general/test_losses.cpp -------------------------------------------------------------------------------- /tests/rl/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/CMakeLists.txt -------------------------------------------------------------------------------- /tests/rl/configs/ddpg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/configs/ddpg.yaml -------------------------------------------------------------------------------- /tests/rl/configs/ppo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/configs/ppo.yaml -------------------------------------------------------------------------------- /tests/rl/configs/sac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/configs/sac.yaml -------------------------------------------------------------------------------- /tests/rl/configs/td3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/configs/td3.yaml -------------------------------------------------------------------------------- /tests/rl/environments.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/environments.h -------------------------------------------------------------------------------- /tests/rl/test_distributions.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_distributions.cpp -------------------------------------------------------------------------------- /tests/rl/test_interface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_interface.cpp -------------------------------------------------------------------------------- /tests/rl/test_off_policy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_off_policy.cpp -------------------------------------------------------------------------------- /tests/rl/test_on_policy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_on_policy.cpp -------------------------------------------------------------------------------- /tests/rl/test_replay_buffer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_replay_buffer.cpp -------------------------------------------------------------------------------- /tests/rl/test_rollout_buffer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/rl/test_rollout_buffer.cpp -------------------------------------------------------------------------------- /tests/supervised/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/CMakeLists.txt -------------------------------------------------------------------------------- /tests/supervised/configs/missing_loss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/missing_loss.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/missing_opt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/missing_opt.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/mlp.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/mlp2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/mlp2.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/mlp2_gradacc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/mlp2_gradacc.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/mlp3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/mlp3.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/torchscript.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/torchscript.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/torchscript_multiarg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/torchscript_multiarg.yaml -------------------------------------------------------------------------------- /tests/supervised/configs/torchscript_multiarg_extra.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/configs/torchscript_multiarg_extra.yaml -------------------------------------------------------------------------------- /tests/supervised/scripts/setup_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/scripts/setup_tests.py -------------------------------------------------------------------------------- /tests/supervised/test_checkpoint.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/test_checkpoint.cpp -------------------------------------------------------------------------------- /tests/supervised/test_distributed_training.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/test_distributed_training.cpp -------------------------------------------------------------------------------- /tests/supervised/test_training.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/supervised/test_training.cpp -------------------------------------------------------------------------------- /tests/test_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA/TorchFort/HEAD/tests/test_utils.h --------------------------------------------------------------------------------