Content pfp
Content
@
https://warpcast.com/~/channel/llm
0 reply
0 recast
0 reaction

christopher pfp
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 pfp
draftcode
@draftcode
This is cool ! Iโ€™m gonna take a look
0 reply
0 recast
1 reaction

Muhammad Ali pfp
Muhammad Ali
@muhammadalii
JAX is next-levelโ€”functional, fast, and runs anywhere seamlessly!
0 reply
0 recast
0 reaction