PyPI page
Home page
Author:
None
Summary:
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning
Latest version:
0.11.1
Required dependencies:
brax
|
cloudpickle
|
dm-meltingpot
|
jax
|
numpy
|
packaging
|
pyvers
|
tensordict
|
torch
Optional dependencies:
accelerate
|
autoflake
|
bitsandbytes
|
datasets
|
dm_control
|
einops
|
flash-attn
|
gymnasium
|
h5py
|
huggingface_hub
|
hydra-core
|
hydra-submitit-launcher
|
immutabledict
|
langdetect
|
minari
|
moviepy
|
mujoco
|
nltk
|
open_spiel
|
pandas
|
peft
|
pettingzoo
|
pillow
|
playwright
|
pre-commit
|
procgen
|
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
|
tensorboard
|
torch
|
torchsnapshot
|
torchvision
|
tqdm
|
transformers
|
vllm
|
vmas
|
wandb
|
xformers
Downloads last day:
36,287
Downloads last week:
83,309
Downloads last month:
623,381