jax
tensorflow
einops
flax
optax
orbax-checkpoint
wandb
numpy
h5py
ml-collections
scikit-fem
scipy
matplotlib