jax-ml/jax

jax-ml/jax: jax-ml/jax may be useful when its README and activity match a concrete workflow.. Use it for Misc. Medium: useful for prototypes or internal workflo