4 ms·
There's a JAX for AI/LM too https://github.com/jax-ml/jax https://github.com/jax-ml/jax but yeah no idea which the OP meant
by tough 1y ago
There's a JAX for AI/LM too
https://github.com/jax-ml/jax https://github.com/jax-ml/jax
but yeah no idea which the OP meant