PyPI page
Author:
NVIDIA CORPORATION & AFFILIATES
License:
Apache-2.0
Summary:
A simple Ray + vLLM + FSDP2 (AutoTP/EP/CP) stack for SFT and RL.
Latest version:
0.1.12
Required dependencies:
accelerate
|
aiohttp
|
datasets
|
einops
|
grpcio
|
huggingface-hub
|
jsonlines
|
nemo-automodel
|
optree
|
packaging
|
peft
|
pylatexenc
|
pynvml
|
ray
|
sympy
|
tensorboard
|
tokenizers
|
torch
|
torchdata
|
torchmetrics
|
tqdm
|
transformers
|
vllm
|
vllm-router
|
wandb
|
wheel
Optional dependencies:
flash-attn
|
vllm
Downloads last day:
3
Downloads last week:
105
Downloads last month:
405
7-day smoothing