JAX & XLA Mental Model
Pure functions, jit/grad/vmap, pytrees, NNX split/merge, and JAX to HLO to device.
0 / 8 solved
Question Difficulty Success
Learning Resources
Up next
8 of 8 questions here are still unsolved.
Pure functions, jit/grad/vmap, pytrees, NNX split/merge, and JAX to HLO to device.
8 of 8 questions here are still unsolved.