Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 18, 2025

• Merge pull request #5149 from IvyZX:release

  Show description for d179184
  

  Flax Authors

  committed
  d179184

  Copy full SHA for d179184

• Release 0.12.2

  IvyZX

  committed
  6d36623

  Copy full SHA for 6d36623

• Merge pull request #5144 from google:fix-ci-5143

  Show description for 9213cfd
  

  Flax Authors

  committed
  9213cfd

  Copy full SHA for 9213cfd

• Temporary fix for failing CI

  Show description for de85317
  vfdev-5

  committed
  de85317

  Copy full SHA for de85317

Commits on Dec 11, 2025

• Merge pull request #5102 from samanklesaria:output_sharding

  Show description for ef78d65
  

  Flax Authors

  committed
  ef78d65

  Copy full SHA for ef78d65

• Merge pull request #5093 from ibbyml:nnx-sdpa-parity

  Show description for be1db78
  

  Flax Authors

  committed
  be1db78

  Copy full SHA for be1db78

• Fix nits

  samanklesaria

  committed
  6654420

  Copy full SHA for 6654420

Commits on Dec 9, 2025

• Copybara import of the project:

  Show description for 3139e96
  

  Flax Team

  committed
  3139e96

  Copy full SHA for 3139e96

• Merge pull request #5057 from jorisSchaller:feat/nnx_pooling

  Show description for de84eb1
  

  Flax Authors

  committed
  de84eb1

  Copy full SHA for de84eb1

• remove fingerprint

  Show description for b8aa99d
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  b8aa99d

  Copy full SHA for b8aa99d

• remove abstracted_axes from nnx.jit

  Show description for b3e9b44
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  b3e9b44

  Copy full SHA for b3e9b44

• [jax:benchmarks] Add tracing/lowering benchmarks for a few flax examples.

  Show description for 2204c82
  

  danielsuo

  authored and

  Flax Authors

  committed
  2204c82

  Copy full SHA for 2204c82

Commits on Dec 5, 2025

• [flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit

  Show description for 872d50b
  

  danielsuo

  authored and

  Flax Authors

  committed
  872d50b

  Copy full SHA for 872d50b

• [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.

  Show description for 881b8ed
  

  danielsuo

  authored and

  Flax Authors

  committed
  881b8ed

  Copy full SHA for 881b8ed

• Refactor pooling operation

  Show description for 004b932
  

  jorisSchaller

  authored and

  vfdev-5

  committed
  004b932

  Copy full SHA for 004b932

Commits on Dec 4, 2025

• Avoid passing `concrete` argument to `jax.remat`

  Show description for ec85cdd
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  ec85cdd

  Copy full SHA for ec85cdd

Commits on Dec 3, 2025

• [flax:examples:ppo] Fix some linter / import issues. #jax-fixit

  Show description for 59cfd99
  

  danielsuo

  authored and

  Flax Authors

  committed
  59cfd99

  Copy full SHA for 59cfd99

Commits on Dec 2, 2025

• [flax:examples:sst2] Fix notebook error.

  Show description for 5a3d148
  

  danielsuo

  authored and

  Flax Authors

  committed
  5a3d148

  Copy full SHA for 5a3d148

• Merge pull request #5094 from thijs-vanweezel:filters

  Show description for a5af24c
  

  Flax Authors

  committed
  a5af24c

  Copy full SHA for a5af24c

Commits on Dec 1, 2025

• [flax:examples:sst2] Fix pytype errors.

  Show description for 6ac5a78
  

  danielsuo

  authored and

  Flax Authors

  committed
  6ac5a78

  Copy full SHA for 6ac5a78

• [flax:examples:gemma] Fixing linter errors.

  Show description for 324f4b4
  

  danielsuo

  authored and

  Flax Authors

  committed
  324f4b4

  Copy full SHA for 324f4b4

• [flax:examples:vae] Small linter fixes.

  Show description for 88d7497
  

  danielsuo

  authored and

  Flax Authors

  committed
  88d7497

  Copy full SHA for 88d7497

• [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb.

  Show description for 9ed59e2
  

  danielsuo

  authored and

  Flax Authors

  committed
  9ed59e2

  Copy full SHA for 9ed59e2

• [flax:examples:wmt] Small linter fixes.

  Show description for c29b6d6
  

  danielsuo

  authored and

  Flax Authors

  committed
  c29b6d6

  Copy full SHA for c29b6d6

Commits on Nov 19, 2025

• Merge pull request #5095 from google:jax-minver-0.8.1

  Show description for 697f4e5
  

  Flax Authors

  committed
  697f4e5

  Copy full SHA for 697f4e5

• Add out_sharding argument to call methods for standard layers

  samanklesaria

  committed
  1f61d6b

  Copy full SHA for 1f61d6b

• update jax minver to 0.8.1

  cgarciae

  committed
  b424c49

  Copy full SHA for b424c49

• hint that key is cast to string

  thijs-vanweezel

  committed
  a7bb109

  Copy full SHA for a7bb109

• test pathcontains with exact=false

  thijs-vanweezel

  committed
  12b2e42

  Copy full SHA for 12b2e42

• allow substring matching

  thijs-vanweezel

  committed
  3d34055

  Copy full SHA for 3d34055

Commits on Nov 18, 2025

• Add is_causal mask argument and tests

  ibbyml

  committed
  9bab148

  Copy full SHA for 9bab148

• Merge pull request #5097 from google:fix-main-4

  Show description for bee81ed
  

  Flax Authors

  committed
  bee81ed

  Copy full SHA for bee81ed

• Fixed deprecated .value usage failing CI tests

  vfdev-5

  committed
  00cf44e

  Copy full SHA for 00cf44e

Commits on Nov 17, 2025

• Merge pull request #5084 from samanklesaria:variable_eq

  Show description for 09537f4
  

  Flax Authors

  committed
  09537f4

  Copy full SHA for 09537f4

• Merge pull request #5052 from google:deprecate-value-2

  Show description for b7e66b7
  

  Flax Authors

  committed
  b7e66b7

  Copy full SHA for b7e66b7

Pagination

Previous

Next