grain == 0.2.18
datasets >= 3.6.0, < 4.8.0
# PyArrow 26 requires NumPy 2; PyTorch CI also covers NumPy 1.x.
pyarrow < 26
tensorboard
wandb
tokenizers >= 0.15.0
renderers == 0.1.12
safetensors
einops
pillow
spmd_types == 0.3.1
attn-gym[linear,cudnn]==0.0.17
torch_remat == 0.2.2
torch_checkpointing == 0.1.1
