AIbase
Product LibraryTool NavigationMCP

jimm

Public

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)

Creat2025-06-05T15:11:31
Update2025-06-18T07:54:46
https://pythoncrazy.github.io/jimm/
2
Stars
0
Stars Increase