PyPI page
Home page
Author:
None
Summary:
Cosmos-RL is a flexible and scalable Reinforcement Learning framework specialized for Physical AI applications.
Latest version:
0.4.3
Required dependencies:
accelerate
|
blobfile
|
boto3
|
click
|
cloudpickle
|
datasets
|
diffusers
|
fastapi
|
kernels
|
liger_kernel
|
math_verify
|
modelscope
|
msgpack
|
nest_asyncio
|
nvidia-ml-py
|
nvidia-nccl-cu12
|
peft
|
pybind11
|
pydantic
|
pynvml
|
pyyaml
|
qwen_vl_utils
|
redis
|
rich
|
strenum
|
tensordict
|
toml
|
torch
|
torchao
|
torchvision
|
transformers
|
uvicorn
|
zmq
Optional dependencies:
aioboto3
|
aiobotocore
|
bddl
|
better_profanity
|
cosmos_rl
|
decord
|
easydict
|
einops
|
flash-attn
|
future
|
gym
|
gymnasium
|
hydra-core
|
imageio
|
imageio-ffmpeg
|
libero
|
math-verify
|
matplotlib
|
nltk
|
numpy
|
opencv-python
|
opencv-python-headless
|
retinaface-py
|
robomimic
|
robosuite
|
sentencepiece
|
termcolor
|
thop
|
timm
|
transformers
|
trimesh
|
vllm
|
wandb
|
webdataset
Downloads last day:
7
Downloads last week:
136
Downloads last month:
638