├── .coveragerc ├── .gitignore ├── LICENSE ├── NOTICE ├── README.md ├── config ├── offline_baselines_jax ├── __init__.py ├── common │ ├── __init__.py │ ├── base_class.py │ ├── buffers.py │ ├── jax_layers.py │ ├── off_policy_algorithm.py │ ├── policies.py │ ├── preprocessing.py │ ├── save_util.py │ ├── type_aliases.py │ ├── utils.py │ └── vec_env │ │ ├── __init__.py │ │ ├── base_vec_env.py │ │ ├── dummy_vec_env.py │ │ ├── stacked_observations.py │ │ ├── subproc_vec_env.py │ │ ├── util.py │ │ ├── vec_check_nan.py │ │ ├── vec_extract_dict_obs.py │ │ ├── vec_frame_stack.py │ │ ├── vec_monitor.py │ │ ├── vec_normalize.py │ │ ├── vec_transpose.py │ │ └── vec_video_recorder.py ├── cql │ ├── __init__.py │ ├── core.py │ └── cql.py ├── py.typed ├── sac │ ├── __init__.py │ ├── core.py │ ├── policies.py │ └── sac.py ├── td3 │ ├── __init__.py │ ├── core.py │ ├── policies.py │ └── td3.py └── version.txt └── setup.py /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/.coveragerc -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/README.md -------------------------------------------------------------------------------- /config: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /offline_baselines_jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/__init__.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /offline_baselines_jax/common/base_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/base_class.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/buffers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/buffers.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/jax_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/jax_layers.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/off_policy_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/off_policy_algorithm.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/policies.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/preprocessing.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/save_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/save_util.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/type_aliases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/type_aliases.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/utils.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/__init__.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/base_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/base_vec_env.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/dummy_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/dummy_vec_env.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/stacked_observations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/stacked_observations.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/subproc_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/subproc_vec_env.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/util.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_check_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_check_nan.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_extract_dict_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_extract_dict_obs.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_frame_stack.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_monitor.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_normalize.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_transpose.py -------------------------------------------------------------------------------- /offline_baselines_jax/common/vec_env/vec_video_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/common/vec_env/vec_video_recorder.py -------------------------------------------------------------------------------- /offline_baselines_jax/cql/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/cql/__init__.py -------------------------------------------------------------------------------- /offline_baselines_jax/cql/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/cql/core.py -------------------------------------------------------------------------------- /offline_baselines_jax/cql/cql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/cql/cql.py -------------------------------------------------------------------------------- /offline_baselines_jax/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /offline_baselines_jax/sac/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/sac/__init__.py -------------------------------------------------------------------------------- /offline_baselines_jax/sac/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/sac/core.py -------------------------------------------------------------------------------- /offline_baselines_jax/sac/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/sac/policies.py -------------------------------------------------------------------------------- /offline_baselines_jax/sac/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/sac/sac.py -------------------------------------------------------------------------------- /offline_baselines_jax/td3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/td3/__init__.py -------------------------------------------------------------------------------- /offline_baselines_jax/td3/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/td3/core.py -------------------------------------------------------------------------------- /offline_baselines_jax/td3/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/td3/policies.py -------------------------------------------------------------------------------- /offline_baselines_jax/td3/td3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/offline_baselines_jax/td3/td3.py -------------------------------------------------------------------------------- /offline_baselines_jax/version.txt: -------------------------------------------------------------------------------- 1 | 1.0.0 -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jsw7460/sb3_jax/HEAD/setup.py --------------------------------------------------------------------------------