Content
@
https://warpcast.com/~/channel/llm
0 reply
0 recast
0 reaction
christopher
@christopher
JAX is pretty sick. Functional programming, automatic differentiation, JIT, vmap + pmap, and TPU/GPU acceleration with the same code that runs on CPU ๐ฎโ๐จ https://docs.jax.dev/en/latest/
2 replies
1 recast
7 reactions
draftcode
@draftcode
This is cool ! Iโm gonna take a look
0 reply
0 recast
1 reaction
Muhammad Ali
@muhammadalii
JAX is next-levelโfunctional, fast, and runs anywhere seamlessly!
0 reply
0 recast
0 reaction