jax>=0.9.0
jaxlib>=0.9.0
matplotlib
numpy
pyvista
Rtree
trimesh
orbax-checkpoint
h5py
pytest
