
[jax]
optax
flax
jax

[testing]
pytest
pytest-cov

[torch]
torch
bitsandbytes
accelerate
