Change the repository type filter
All
Repositories list
3 repositories
jimm
PublicJAX 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)locamage.github.io
Public.github
Public