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 :)
-
Notifications
You must be signed in to change notification settings - Fork 0
kelechi-c/play_jax
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
sandbox for different JAX projects
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published