JAX

I need to learn JAX, originally this was recommended by Ryan Erlich.

Related

• FLAX