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
zhangxueyou
@zhangxueyou
First-rate
0 reply
0 recast
0 reaction