jax
jaxlib
torch

[dev]
ruff
pre-commit
