Optax is a gradient processing and optimization library for JAX.
active 2023-09-08 → 2026-08-06 (UTC)
Activity over time
Daily event counts in the loaded window
Line chart, 1064 days from 2023-09-08 to 2026-08-06. Pushes: 2,285 total, peak 38 in a day. Pull requests: 1,214 total, peak 16 in a day. Issues: 397 total, peak 12 in a day. Comments: 2,120 total, peak 29 in a day. Stars: 838 total, peak 6 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] | 3.1K | 2.2K | 837 | 0 |
| vroulet | 693 | 0 | 25 | 384 |
| fabianp | 640 | 13 | 81 | 344 |
| rdyro | 490 | 25 | 25 | 306 |
| carlosgmartin | 326 | 0 | 63 | 197 |
| mtthss | 80 | 0 | 7 | 41 |
| google-cla[bot] | 70 | 0 | 0 | 70 |
| rajasekharporeddy | 51 | 0 | 6 | 7 |
| amosyou | 45 | 0 | 5 | 37 |
| evanatyourservice | 43 | 0 | 1 | 29 |
| mathDR | 38 | 0 | 3 | 25 |
| SauravMaheshkar | 32 | 0 | 3 | 26 |
| Tomas542 | 31 | 0 | 5 | 23 |
| tinker495 | 29 | 0 | 1 | 16 |
| leochlon | 28 | 0 | 4 | 19 |
| emilyfertig | 28 | 0 | 0 | 16 |
| mmhamdy | 28 | 0 | 4 | 21 |
| MaanasArora | 26 | 0 | 2 | 16 |
| mblondel | 25 | 0 | 2 | 13 |
| aymuos15 | 21 | 0 | 2 | 15 |
Recent activity
Latest issues, pull requests and releases
- Pull request#1718irhyl2026-06-27 22:25
- Pull request#1716codewithfourtix2026-06-27 08:11
- Issue comment#1697google-cla[bot]2026-06-11 03:02contrib: add MARS optimizer (arXiv:2411.10438)
- Pull request#1691copybara-service[bot]2026-06-05 18:35
- Pull request#1682copybara-service[bot]2026-05-26 21:23
- Pull request#1676copybara-service[bot]2026-05-18 18:10
- Pull request#1670copybara-service[bot]2026-05-03 18:01
- Issue comment#1668google-cla[bot]2026-04-25 09:29Improve microbatching support for positional args passed by name and other variations of positional/kwargs.
- Pull request#1665copybara-service[bot]2026-04-24 19:34
- Pull request#1659armbrusl2026-04-10 12:47
- Issue#1658rajasekharporeddy2026-04-10 06:49Add Self-Scaled Broyden (SSBroyden) and SSBFGS optimizers
- Issue#1656impacterai2026-04-02 22:36Add Repurpose AI — AI content repurposing tool
- Issue comment#1643dylantirandaz2026-03-24 00:46Extend muon to support mask_callable weight dimension numbers
- Pull request#1646dylantirandaz2026-03-23 06:45
- Issue#1645blackblitz2026-03-23 06:44Missing type annotation for sigmoid_binary_cross_entropy
- Releaseemilyfertig2026-03-20 23:29Optax 0.2.8
- Pull request#1637copybara-service[bot]2026-03-20 22:53
- Pull request#1635rdyro2026-03-20 05:02
- Issue comment#1604rdyro2026-03-11 04:57Adahessian optimizer and refactor Hutchinson estimator as shared utility to be used by Adahessian, Sophia optimizer
- Issue#1622carlesoctav2026-03-10 03:39Microbatch API: Outside Loop Initialization Appears to Double Memory Usage
- Issue comment#1622carlesoctav2026-03-10 03:39Microbatch API: Outside Loop Initialization Appears to Double Memory Usage
- Issue#1621rajasekharporeddy2026-03-09 04:27Add PolarExpress Variant to Muon
- Issue comment#1613MarcMachaczek2026-03-08 05:55Add PolarExpress variant to Muon optimizer
- Issue comment#1613NoahAmsel2026-03-07 21:11Add PolarExpress variant to Muon optimizer
- Issue comment#1612rdyro2026-03-05 06:12Add NorMuon optimizer
Totals cover only the window loaded into ClickHouse and count events, not GitHub's lifetime totals — 838 stars here means stars gained during the window, not the repo's star count.