skyrl

PyPI page
Summary: Unified API for training and inference
Latest version: 0.4.0
Required dependencies: cloudpathlib | datasets | hf-transfer | peft | pillow | rich | safetensors | tokenizers | transformers | typer | xxhash | zstandard
Optional dependencies: accelerate | aiosqlite | alembic | asyncpg | causal-conv1d | cloudpathlib | datasets | debugpy | fast-hadamard-transform | fastapi | flash-attn | flash-attn-4 | flash-linear-attention | flashinfer-cubin | flashinfer-jit-cache | flashinfer-python | flax | func-timeout | gcsfs | griffe2md | harbor | hf-transfer | hydra-core | jax | jaxtyping | litellm | loguru | mamba-ssm | megatron-bridge | megatron-core | mini-swe-agent | mkdocs | mkdocs-material | mkdocstrings | mooncake-transfer-engine-cuda13 | ninja | nixl | nvidia-modelopt | nvidia-resiliency-ext | omegaconf | optax | orjson | peft | polars | pre-commit | psycopg2-binary | pybase64 | pybind11 | pymdown-extensions | pytest | pytest-asyncio | pytest-forked | ray | s3fs | s5cmd | setuptools | skycap | skyrl | skyrl-gym | sqlalchemy | sqlmodel | tensorboard | tensordict | tinker | torch | torchdata | torchvision | tqdm | transformer-engine | transformer-engine-cu13 | transformer-engine-torch | ty | uvicorn | vllm | vllm-router | wandb

Downloads last day: 3
Downloads last week: 23
Downloads last month: 210
7-day smoothing