chanfig
gitpython

[jax]
jax
flax

[tensorflow]
tensorflow

[torch]
torch
accelerate
