Ice River
@iceriver
JAX is pretty sick. Functional programming, automatic differentiation, JIT, vmap + pmap, and TPU/GPU acceleration with the same code that runs on CPU 😮💨
0 reply
0 recast
3 reactions
Maxine
@maxine021
Insightful
0 reply
0 recast
0 reaction
