Dependency Graph for optax 0.0.3