Commits

Branch selector

 main

User selector

danielsuo

Datepicker

All time

Commit History

Commits on Jan 6, 2026

• [pmap] Avoid degraded performance under the new `jax.pmap`.

  Show description for 5d90851
  

  danielsuo

  authored and

  Flax Authors

  committed
  5d90851

  Copy full SHA for 5d90851

Commits on Dec 24, 2025

• [pmap] Update `jax_utils.unreplicate` for performant use when `jax.config.jax_pmap_shmap_merge` is `True`.

  Show description for 3f2ff40
  

  danielsuo

  authored and

  Flax Authors

  committed
  3f2ff40

  Copy full SHA for 3f2ff40

Commits on Dec 9, 2025

• [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

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

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 8, 2024

• Fix PRNG handling in `nn.jit` under `nn.scan`.

  Show description for 0f631a2
  

  danielsuo

  authored and

  Flax Authors

  committed
  0f631a2

  Copy full SHA for 0f631a2

Commits on Jun 5, 2020

• Make clear that we are undoing git apply if tests fail in check_howto_diffs.sh

  danielsuo

  committed
  b8610d7

  Copy full SHA for b8610d7

• Improve error message in check_howto_diffs.sh

  danielsuo

  committed
  7c649dd

  Copy full SHA for 7c649dd

Commits on Jun 4, 2020

• Move howto index page

  danielsuo

  committed
  f62618c

  Copy full SHA for f62618c

• Add new howto docs

  danielsuo

  committed
  7d5fdc3

  Copy full SHA for 7d5fdc3

• Move howto docs to their own directory

  danielsuo

  committed
  2774dcb

  Copy full SHA for 2774dcb

• Add check for howto documentation file

  danielsuo

  committed
  21c396e

  Copy full SHA for 21c396e

• Add checkingpointing page

  danielsuo

  committed
  b6cacae

  Copy full SHA for b6cacae

• Give each howto its own page

  danielsuo

  committed
  b55e186

  Copy full SHA for b55e186

• Update ensembling diff

  danielsuo

  committed
  d14de5a

  Copy full SHA for d14de5a

Commits on Jun 3, 2020

• Update distributed-training howto

  danielsuo

  committed
  782a9ae

  Copy full SHA for 782a9ae

• Update checkpointing howto

  danielsuo

  committed
  77efe0d

  Copy full SHA for 77efe0d

• Merge commit to update in line with master

  danielsuo

  committed
  a93323b

  Copy full SHA for a93323b

• Fix up per @avital comments

  danielsuo

  committed
  a06479f

  Copy full SHA for a06479f

Commits on May 29, 2020

• Merge branch 'master' into patch-1

  danielsuo

  authored
  ae45f80

  Copy full SHA for ae45f80

• Add *.rej to .gitignore

  danielsuo

  committed
  04bcb23

  Copy full SHA for 04bcb23

• Merge pull request #298 from danielsuo/dsuo/update-tensorboard-tests

  Show description for c2a6346
  danielsuo

  authored
  c2a6346

  Copy full SHA for c2a6346

• Update tensorboard tests

  danielsuo

  committed
  4aec10d

  Copy full SHA for 4aec10d

Commits on May 28, 2020

• Fix indent in checkpointing.diff

  danielsuo

  committed
  2bcc42c

  Copy full SHA for 2bcc42c

• Fix typo in distributed-training.diff

  danielsuo

  committed
  b0ca3b5

  Copy full SHA for b0ca3b5

• Remove dead comment

  danielsuo

  committed
  a51bcb9

  Copy full SHA for a51bcb9

• Clean up check_howto_diffs

  danielsuo

  committed
  edf385e

  Copy full SHA for edf385e

• Pasdf

  danielsuo

  committed
  e36c87a

  Copy full SHA for e36c87a

Pagination

Previous

Next