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
0 reaction
Tudou
@tudouer
Happy with it
0 reply
0 recast
0 reaction