mlx>=0.31
numpy
torch
safetensors
rwkv-metal>=0.1.3

[dev]
pytest
