A JAX research toolkit for visualizing neural networks.
active 2024-04-17 → 2026-08-15 (UTC)
Activity over time
Daily event counts in the loaded window
Line chart, 851 days from 2024-04-17 to 2026-08-15. Pushes: 275 total, peak 25 in a day. Pull requests: 168 total, peak 15 in a day. Issues: 46 total, peak 4 in a day. Comments: 139 total, peak 11 in a day. Stars: 1,740 total, peak 337 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 |
|---|---|---|---|---|
| copybara-service[bot] | 333 | 206 | 126 | 0 |
| danieldjohnson | 193 | 69 | 28 | 65 |
| JEM-Mosig | 25 | 0 | 1 | 18 |
| amifalk | 20 | 0 | 1 | 11 |
| AntoinePlumerault | 19 | 0 | 0 | 11 |
| femtomc | 16 | 0 | 0 | 7 |
| guxm2021 | 9 | 0 | 2 | 5 |
| srush | 8 | 0 | 0 | 3 |
| ealt | 6 | 0 | 0 | 3 |
| hrbigelow | 5 | 0 | 0 | 2 |
| google-cla[bot] | 5 | 0 | 0 | 5 |
| Fedzbar | 3 | 0 | 3 | 0 |
| samsja | 3 | 0 | 1 | 2 |
| kochkov92 | 3 | 0 | 1 | 2 |
| Samarendra109 | 2 | 0 | 0 | 1 |
| pitmonticone | 2 | 0 | 1 | 1 |
| shoyer | 2 | 0 | 1 | 0 |
| Jammf | 2 | 0 | 2 | 0 |
| IvyZX | 1 | 0 | 0 | 1 |
| theicehouse | 1 | 0 | 0 | 0 |
Recent activity
Latest issues, pull requests and releases
- Pull request#141sylvesterkaczmarek2026-08-15 15:54
- Issue#131amifalk2025-12-03 22:42`nmap` does not respect enclosing `vmap`
- Pull request#130Jammf2025-10-17 22:01
- Pull request#129Jammf2025-10-16 20:55
- Issue#126neverix2025-08-01 21:31Passing extra arguments to gradient transformations
- Issue comment#107Samarendra1092025-07-23 11:03Initializer for Penzai's Linear layer does not support `jax.nn.initializers`
- Issue comment#120danieldjohnson2025-06-28 22:08Feature Request: Add convolution layers
- Issue comment#120guxm20212025-06-24 21:49Feature Request: Add convolution layers
- Issue comment#120guxm20212025-06-24 17:15Feature Request: Add convolution layers
- Issue comment#120AntoinePlumerault2025-06-22 16:29Feature Request: Add convolution layers
- Issue comment#120AntoinePlumerault2025-06-22 16:21Feature Request: Add convolution layers
- Issue#115danieldjohnson2025-06-22 05:44Conversion from pretrained HuggingFace models
- Issue#112danieldjohnson2025-06-22 05:44Conversion of a LlamaForCausalLM does not support `use_cache` and `_name_or_path`
- Pull request#125danieldjohnson2025-06-22 05:44
- Issue comment#125danieldjohnson2025-06-22 05:44Handle missing attributes in HuggingFace conversion functions
- Pull request#125danieldjohnson2025-06-22 05:28
- Issue#117danieldjohnson2025-06-22 04:52BUG: potential tracer leak in _jitted_nmapped_getitem?
- Pull request#124danieldjohnson2025-06-22 04:52
- Pull request#124danieldjohnson2025-06-22 04:43
- Issue comment#120danieldjohnson2025-06-21 16:43Feature Request: Add convolution layers
- Pull request#119danieldjohnson2025-06-20 05:36
- Issue comment#123danieldjohnson2025-06-20 05:08ValueError: Cannot serialize PyTreeDef containing an unregistered type `<class 'penzai.core.named_axes.NamedArray'>`.
- Issue comment#122danieldjohnson2025-06-20 05:01LayerStack.from_sublayer_builder temporarily uses about twice the memory
- Issue comment#122JEM-Mosig2025-06-16 10:07LayerStack.from_sublayer_builder temporarily uses about twice the memory
- Issue comment#122JEM-Mosig2025-06-16 09:44LayerStack.from_sublayer_builder temporarily uses about twice the memory
Totals cover only the window loaded into ClickHouse and count events, not GitHub's lifetime totals — 1,740 stars here means stars gained during the window, not the repo's star count.