torchrl

PyPI page
Home page
Author: None
Summary: A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
Latest version: 0.12.0
Required dependencies: brax | cloudpickle | dm-meltingpot | jax | numpy | packaging | procgen | pyvers | tensordict | torch
Optional dependencies: accelerate | autoflake | bitsandbytes | botorch | datasets | dm_control | einops | flash-attn | genesis-world | gpytorch | gymnasium | h5py | huggingface_hub | hydra-core | hydra-submitit-launcher | immutabledict | langdetect | minari | moviepy | mujoco | nltk | open_spiel | pandas | peft | pettingzoo | pillow | playwright | pre-commit | protobuf | psutil | pytest | pytest-asyncio | pytest-benchmark | pytest-cov | pytest-error-for-skips | pytest-forked | pytest-instafail | pytest-isolate | pytest-mock | pytest-random-order | pytest-repeat | pytest-rerunfailures | pytest-timeout | pyyaml | ray | requests | safetensors | scikit-learn | scipy | sentencepiece | sglang | tensorboard | torch | torchcodec | torchrl | torchsnapshot | torchvision | tqdm | transformers | vllm | vmas | wandb | xformers

Downloads last day: 54,822
Downloads last week: 305,950
Downloads last month: 1,310,233