numpy
polars
scipy

[bpe]
rustbpe
tiktoken

[dev]
pytest>=8.0
rustbpe
tiktoken
torch

[metrics]
torch
