.gitignore
LICENSE
Readme.md
install_venv.sh
into_venv.sh
setup.py
examples/run_rl.py
examples/configs/sac_config.py
jaxdl/__init__.py
jaxdl.egg-info/PKG-INFO
jaxdl.egg-info/SOURCES.txt
jaxdl.egg-info/dependency_links.txt
jaxdl.egg-info/requires.txt
jaxdl.egg-info/top_level.txt
jaxdl/nlp/__init__.py
jaxdl/nlp/utils.py
jaxdl/nn/__init__.py
jaxdl/nn/dnn/__init__.py
jaxdl/nn/dnn/mlp.py
jaxdl/nn/gnn/__init__.py
jaxdl/rl/__init__.py
jaxdl/rl/agents/__init__.py
jaxdl/rl/agents/sac/__init__.py
jaxdl/rl/agents/sac/actor_fns.py
jaxdl/rl/agents/sac/critic_fns.py
jaxdl/rl/agents/sac/sac.py
jaxdl/rl/agents/sac/temperature_fns.py
jaxdl/rl/environments/__init__.py
jaxdl/rl/environments/env_wrappers.py
jaxdl/rl/environments/episode_monitor.py
jaxdl/rl/networks/__init__.py
jaxdl/rl/networks/actor_nets.py
jaxdl/rl/networks/critic_nets.py
jaxdl/rl/networks/temperature_nets.py
jaxdl/rl/utils/__init__.py
jaxdl/rl/utils/commons.py
jaxdl/rl/utils/replay_buffer.py
jaxdl/utils/__init__.py
jaxdl/utils/commons.py
tests/__init__.py
tests/nlp/test_utils.py
tests/nn/__init__.py
tests/nn/test_mlp.py
tests/nn/test_transformer.py
tests/rl/__init__.py
tests/rl/test_agents.py
tests/rl/test_envs.py
tests/rl/test_networks.py
tests/rl/test_utils.py
utils/learning_curves/Ant-v2.png
utils/learning_curves/HalfCheetah-v2.png
utils/learning_curves/Humanoid-v2.png
utils/learning_curves/Pendulum-v1.png
utils/learning_curves/Reacher-v2.png
utils/learning_curves/generate_plots.py