torch
transformers>=5
flash-linear-attention
numpy
huggingface_hub
jinja2

[games]
gymnasium
gym-super-mario-bros
nes-py
minigrid
ale-py
imageio

[gguf]
llama-cpp-python<0.4,>=0.3.35

[metal]

[metal:sys_platform == "darwin" and platform_machine == "arm64"]
mlx>=0.20

[serve]
fastapi
uvicorn[standard]
httpx

[train]
datasets
accelerate
scikit-learn
pillow
