requirements.txt

# pytorch
torch>=2.6.0
torchvision>=0.21.0
torchaudio>=2.6.0

# jax and friends
jax>=0.5.0
optax>=0.2.4
equinox>=0.11.11
jaxtyping>=0.2.38
flax>=0.10.3

# huggingface
diffusers>=0.32.2
optimum>=1.24.0
safetensors>=0.5.2
transformers>=4.49.0
accelerate>=1.6.0
huggingface_hub>=0.30.1

# other machine learning
gguf>=0.14.0
einops>=0.8.1
lightning>=2.5.0.post0
scikit-learn>=1.6.1
tensorboard>=2.19.0
torch-tb-profiler>=0.4.3
wandb>=0.19.8
wandb-workspaces>=0.1.12

# data science
numpy>=2.2.3
polars>=1.22.0
pandas>=2.2.3
scipy>=1.15.2

# math
sympy>=1.13.1
mpmath>=1.3.0

# plotting
matplotlib>=3.10.0
seaborn>=0.13.2

# assorted
tomli-w>=1.2.0
pint>=0.24.4
pyarrow>=19.0.0
protobuf>=5.29.3
flatbuffers>=25.2.10
beartype>=0.19.0

# ui
tqdm>=4.67.1
shiny>=1.2.1

# devtools
black>=25.1.0
flake8>=7.1.2
flake8-bugbear>=24.12.12
pytest>=8.3.4
pytest-cov>=6.0.0

# jupyter and spyder
ipykernel>=6.29.5
ipympl>=0.9.6
ipywidgets>=8.1.5
spyder-kernels>=3.0.3

# python deprecations
standard-imghdr~=3.13.0

#other
unidecode