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)
active 2025-06-12 → 2026-06-29 (UTC)
Activity over time
Daily event counts in the loaded window
Line chart, 383 days from 2025-06-12 to 2026-06-29. Pushes: 64 total, peak 8 in a day. Pull requests: 7 total, peak 2 in a day. Issues: 6 total, peak 3 in a day. Comments: 45 total, peak 11 in a day. Stars: 2 total, peak 1 in a day.
- Pushes
- Pull requests
- Issues
- Comments
- Stars
Stars, PRs, issues and forks are under-captured in the later part of this window. GH Archive progressively stopped capturing non-push events during 2026 — −95% or worse by the end of the window. Every series here except Pushes fades for that reason, so a decline above reflects the archive, not this repository. Pushes stay reliable throughout, so read them, and the contributor counts derived from them, as the real signal. Data health has the measurements.
Top contributors
Pushes, PRs, issues, reviews and comments — stars and forks excluded, so this is contribution rather than popularity
| Contributor | Contributions | Pushes | PRs | Comments |
|---|---|---|---|---|
| pythoncrazy | 67 | 47 | 7 | 7 |
| Copilot | 28 | 0 | 0 | 16 |
| gemini-code-assist[bot] | 19 | 0 | 0 | 15 |
| github-actions[bot] | 17 | 17 | 0 | 0 |
| claude[bot] | 7 | 0 | 0 | 7 |
Recent activity
Latest issues, pull requests and releases
- Pull request#77pythoncrazy2026-06-29 00:25
- Issue comment#59pythoncrazy2026-04-19 01:41Add support for using different attention mechanisms via nnx.MultiHeadAttention
- Releasepythoncrazy2026-01-21 07:59v0.2.0: Separate Vision/Text/Overall Models, and better sharding!
- Issue comment#69gemini-code-assist[bot]2026-01-21 07:58fix: fixed the vision_position_ids sharding
- Issue comment#67pythoncrazy2026-01-19 20:39Splash attention
- Issue comment#67gemini-code-assist[bot]2026-01-19 20:27Splash attention
- Issue comment#67pythoncrazy2026-01-19 20:24Splash attention
- Issue comment#66pythoncrazy2026-01-19 20:18feat: add TPU splash attention support via tokamax
- Issue comment#66pythoncrazy2026-01-19 20:15feat: add TPU splash attention support via tokamax
- Issue#61pythoncrazy2025-12-28 07:58Fix jax.distributed.initialize() with jimm
- Pull request#65pythoncrazy2025-12-28 07:58
- Issue comment#65claude[bot]2025-12-28 07:55Distributed initialize fix by changing the RNGs to lazily init in the methods
- Pull request#65pythoncrazy2025-12-28 07:53
- Issue#64pythoncrazy2025-12-28 06:39Export models under jimm rather than jimm.models
- Issue#64pythoncrazy2025-12-28 06:34Export models under jimm rather than jimm.models
- Issue#62pythoncrazy2025-12-27 09:06Add from_config support, for the various classes
- Issue comment#63claude[bot]2025-12-27 07:53feat: added from_config
- Pull request#63pythoncrazy2025-12-27 07:51
- Issue comment#60claude[bot]2025-12-23 07:56Fix int issues
- Issue comment#60claude[bot]2025-12-23 07:21Fix int issues
- Issue#59pythoncrazy2025-12-22 09:09Add support for using different attention mechanisms via nnx.MultiHeadAttention
- Issue comment#58claude[bot]2025-12-22 08:51fix: got the training loop working, with the int issue
- Issue comment#58claude[bot]2025-12-03 21:26fix: got the training loop working, with the int issue
- Pull request#58pythoncrazy2025-12-03 21:25
- Issue comment#16pythoncrazy2025-11-16 22:46add benchmarks
Totals cover only the window loaded into ClickHouse and count events, not GitHub's lifetime totals — 2 stars here means stars gained during the window, not the repo's star count.