3 ms·

I think the intended footnote was accidentally left out. Were you talking about this Python library? https://docs.jax.dev/en/latest/index.html https://docs.jax

I think the intended footnote was accidentally left out. Were you talking about this Python library? https://docs.jax.dev/en/latest/index.html https://docs.jax.dev/en/latest/index.html

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