https://github.com/pythoncrazy/jimm

JAX Image Modeling of Models contains Computer Vision/Vision Language Model implementations in native flax nnx with proper sharding annotations (allowing you to do easy fully sharded data parallel training with native flax nnx/jax)

computer-vision flax jax machine-learning

Last synced: 21 Jan 2026

https://github.com/pythoncrazy/listofreading

My readinglist, inspired by codemicro.

Last synced: 21 Jan 2026