
jax>=0.3.5
pytreeclass >= 0.1.13
kernex >=0.0.7
