jax-ml

Pushing back the limits on numerical computing.