torch
transformers>=5
flash-linear-attention
numpy<2
huggingface_hub

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

[metal]

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

[serve]
fastapi
uvicorn[standard]
httpx

[train]
datasets
accelerate
scikit-learn
pillow
