Dependencies
List of direct dependencies declared by the software package through its manifest.
jax
Version: >=0.4.2
matplotlib
Version:
msgpack
Version:
numpy
Version: >=1.12
optax
Version:
orbax-checkpoint
Version:
pyyaml
Version: >=5.4.1
rich
Version: >=11.1
tensorstore
Version:
typing-extensions
Version: >=4.2