grain==0.2.18
datasets<4.8.0,>=3.6.0
pyarrow<26
tensorboard
wandb
tokenizers>=0.15.0
renderers==0.1.12
safetensors
einops
pillow
spmd_types==0.3.1
attn-gym[cudnn,linear]==0.0.17
torch_remat==0.2.2
torch_checkpointing==0.1.1

[dev]
expecttest>=0.2.0
pytest==7.3.2
pytest-cov
pre-commit
pyrefly==0.45.1
transformers
numpy

[flux]
transformers>=5.10.1
sentencepiece

[rl]
torchmonarch>=0.7.0
opentelemetry-sdk
opentelemetry-exporter-otlp-proto-http
pygtrie
portpicker

[vlm]
av
einops
pillow
torchvision
