Serenstar
@serenstar
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
0 reaction