night
@nightowlc
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
16 reactions
Boluo
@daboluo
Splendid
0 reply
0 recast
0 reaction