Repository navigation

Commits

Commits on Nov 5, 2026
- Align Python 3.12–3.15 classifiers, GitHub Actions CI, and Pyrefly type checking with JAX
  
  IvyZX
  authored and
  Flax Authors
  committed
  71c5b9c
Commits on Nov 3, 2026
- Use `jax.tree_util.is_tree_node` instead of private `_registry` in NNX
  
  IvyZX
  authored and
  Flax Authors
  committed
  b3243f7
Commits on Oct 27, 2026
- Release Flax 0.12.12
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  e937d47
- rollback of breaking change
  
  levskaya
  authored and
  Flax Authors
  committed
  62ad9d1
Commits on Oct 26, 2026
- Merge pull request #5593 from mayuriphad:fix-flatten-dict-sep
  Flax Authors
  committed
  6866a65
- Merge pull request #5592 from Anu27n:fix/nnx-mha-decode-is-causal
  Flax Authors
  committed
  3c3ce3b
- Merge pull request #5582 from jdymitarai:fix/nnx-capture-scan-list
  Flax Authors
  committed
  bf06ba3
- Add test for invalid keys with sep
  
  mayuriphad
  authored and
  samanklesaria
  committed
  8a285ab
- Update flatten_dict docstring
  
  mayuriphad
  authored and
  samanklesaria
  committed
  12e226b
- Fix flatten_dict to raise ValueError on invalid keys with sep
  
  mayuriphad
  authored and
  samanklesaria
  committed
  d2bdce7
- Fix NNX MultiHeadAttention is_causal mask during decode.
  
  Anu27n
  authored and
  samanklesaria
  committed
  c3721a2
- Release Flax 0.12.11
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  899109a
- [nnx] Make `nnx.Sequential` iterable and filter `**kwargs` per layer.
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  7a24e6a
Commits on Oct 24, 2026
- fix(nnx): skip Sequence modules in capture to prevent mixed key types (fixes #5571)
  jdymitarai
  committed
  ce5acb2
Commits on Oct 21, 2026
- Merge pull request #5623 from IvyZX:fix-5577-weightnorm
  Flax Authors
  committed
  479d43d
- Merge pull request #5488 from samanklesaria:multimodal_llm_example
  Flax Authors
  committed
  2e2876d
Commits on Oct 20, 2026
- Store nnx.WeightNorm scales as Param so they are trainable
  IvyZX
  committed
  ae6d283
- Merge remote-tracking branch 'upstream/main' into multimodal_llm_example
  cgarciae
  committed
  3ee9be7
- Merge pull request #5472 from samanklesaria:audio_example
  Flax Authors
  committed
  7a610fa
- Print shapes instead of exact numerics in Embed docstrings
  IvyZX
  committed
  c15ac73
- Fix mypy errors with jax 0.12.0
  IvyZX
  committed
  82ca84e
- Merge remote-tracking branch 'origin/main' into audio_example
  IvyZX
  committed
  c3d172b
Commits on Oct 12, 2026
- Add `capture_noop` option to `Module.perturb`.
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  41deea3
Commits on Oct 8, 2026
- Don't pass linen_meta_type argument when creating AxisMetadata subclasses.
  Flax Team
  committed
  6c9494b
Commits on Oct 7, 2026
- Fix Flax GitHub Actions CI (`pytest` `DeprecationWarning` and `pytype` in `examples/gemma/train.py`).
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  5e60ced
Commits on Oct 6, 2026
- Replace `# pytype: disable` suppressions with `# pyrefly: ignore`
  
  oprypin
  authored and
  Flax Authors
  committed
  81b168d
Commits on Oct 1, 2026
- Add `nnx_sow_capture_noop` config flag and `capture_noop` option to `Module.sow`.
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  6cde9 8BD0 7d
Commits on Sep 25, 2026
- Merge pull request #5518 from chenkuanliao:fix-issue-5512
  Flax Authors
  committed
  4c44124
- Merge pull request #5580 from samanklesaria:fix_mypy_hijax_error
  Flax Authors
  committed
  78abd68
- Fix mypy hijax error
  samanklesaria
  committed
  ce09c08
Commits on Sep 24, 2026
- Release Flax 0.12.10 and use attribute lookup for HiPrim fallback
  
  Cristian Garcia
  authored and
  Flax Authors
  committed
  d1f0d69
Commits on Sep 16, 2026
- suppress pyrefly errors related to JAX scalar types
  
  Jake VanderPlas
  authored and
  Flax Authors
  committed
  4f42ed0
Commits on Sep 15, 2026
- Add sepreformer example
  samanklesaria
  committed
  02d3065
- Add multimodal llm example
  samanklesaria
  committed
  6a414b0
Commits on Aug 21, 2026
- [No functional change] Delete VJPHiPrimitive and replace with HiPrim. This is a simple replace.
  
  yashk2810
  authored and
  Flax Authors
  committed
  01854da

Branch selector

User selector

Datepicker

Commit history

Commits on Nov 5, 2026

Commits on Nov 3, 2026

Commits on Oct 27, 2026

Commits on Oct 26, 2026

Commits on Oct 24, 2026

Commits on Oct 21, 2026

Commits on Oct 20, 2026

Commits on Oct 12, 2026

Commits on Oct 8, 2026

Commits on Oct 7, 2026

Commits on Oct 6, 2026

Commits on Oct 1, 2026

Commits on Sep 25, 2026

Commits on Sep 24, 2026

Commits on Sep 16, 2026

Commits on Sep 15, 2026

Commits on Aug 21, 2026