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
Muhammad Ali
@muhammadalii
JAX is next-level—functional, fast, and runs anywhere seamlessly!
0 reply
0 recast
0 reaction
