About Me
Resume
Blog
Blog Posts
Categories
Deep Learning
|
Agentic AI
|
Full Stack
Deep Learning
My First JAX Contribution: Modernizing SPMD Examples from pmap to jax.jit
January 15, 2025
My First JAX Contribution: Modernizing SPMD Examples from pmap to jax.jit
Read more
Agentic AI
Full Stack