skyrl

PyPI page
Home page
Author: None
Summary: Unified API for training and inference
Latest version: 0.2.0
Required dependencies: causal-conv1d | cloudpathlib | datasets | flash-attn | flash-linear-attention | flashinfer-jit-cache | flashinfer-python | harbor | hf_transfer | jax | mamba-ssm | megatron-bridge | megatron-core | nixl | nvidia-modelopt | peft | pillow | rich | safetensors | tokenizers | torch | torchvision | transformer-engine | transformers | typer | vllm | vllm-router
Optional dependencies: accelerate | aiosqlite | alembic | asyncpg | cloudpathlib | datasets | debugpy | fastapi | flax | func_timeout | griffe2md | hf_transfer | hydra-core | jax | jaxtyping | litellm | loguru | mini-swe-agent | mkdocs | mkdocs-material | mkdocstrings | ninja | omegaconf | optax | peft | polars | pre-commit | psycopg2-binary | pybind11 | pymdown-extensions | pytest | pytest-asyncio | pytest-forked | ray | s3fs | setuptools | skyrl | skyrl-gym | sqlalchemy | sqlmodel | tensorboard | tensordict | tinker | torch | torchdata | tqdm | ty | uvicorn | wandb

Downloads last day: 4
Downloads last week: 69
Downloads last month: 254