Skip to content

kelechi-c/play_jax

Repository files navigation

play_jax

My experiments with JAX

More like my learning sandbox. I am currently porting most of my stuff to JAX (with NNX), so yeah. First thing I ported/rewrote in JAX was the Mobilenet CNN. Next up, diffusion, transformers, etc :)

About

sandbox for different JAX projects

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published