Similar repositories to matthewfeickert/jax-exploration: