Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 18, 2025

• Merge pull request #4716 from google:update-ml-collections

  Show description for dbce607
  

  Flax Authors

  committed
  dbce607

  Copy full SHA for dbce607

• update ml-collections

  cgarciae

  committed
  ccdd29f

  Copy full SHA for ccdd29f

Commits on Apr 16, 2025

• Make fully replicated array before saving checkpoints for examples that use pmap.

  Show description for aa89779
  

  IvyZX

  authored and

  Flax Authors

  committed
  aa89779

  Copy full SHA for aa89779

Commits on Apr 15, 2025

• Merge pull request #4630 from google:optimize-jit

  Show description for 1238a61
  

  Flax Authors

  committed
  1238a61

  Copy full SHA for 1238a61

• Merge pull request #4705 from IvyZX:einsum

  Show description for 6310690
  

  Flax Authors

  committed
  6310690

  Copy full SHA for 6310690

• Add custom einsum op to Einsum()

  IvyZX

  committed
  000ee89

  Copy full SHA for 000ee89

Commits on Apr 14, 2025

• Rollback of attention_bias parameter, because the change overrides the attention bias for injected attention functions.

  Show description for 32ad6c5
  

  Jan Hosang

  authored and

  Flax Authors

  committed
  32ad6c5

  Copy full SHA for 32ad6c5

• Unit tests for `attention_bias` parameter to `MultiHeadDotProductAttention`. Add parameter to all overloads to make pytype happy.

  Show description for 5bba7b4
  

  Jan Hosang

  authored and

  Flax Authors

  committed
  5bba7b4

  Copy full SHA for 5bba7b4

• [nnx] refactor GraphDef

  cgarciae

  committed
  8cb8a7a

  Copy full SHA for 8cb8a7a

Commits on Apr 11, 2025

• Add `attention_bias` parameter to `MultiHeadDotProductAttention`.

  Show description for 2fae1ba
  

  Jan Hosang

  authored and

  Flax Authors

  committed
  2fae1ba

  Copy full SHA for 2fae1ba

Commits on Apr 10, 2025

• Merge pull request #4627 from google:remove-deprecated

  Show description for aa9d572
  

  Flax Authors

  committed
  aa9d572

  Copy full SHA for aa9d572

Commits on Apr 9, 2025

• Merge pull request #4686 from kaixih:support_fp8_einsum

  Show description for fe0b35d
  

  Flax Authors

  committed
  fe0b35d

  Copy full SHA for fe0b35d

• Format

  kaixih

  committed
  01cdcc7

  Copy full SHA for 01cdcc7

Commits on Apr 8, 2025

• Update the doc

  kaixih

  committed
  76f89b1

  Copy full SHA for 76f89b1

• Update the fp8 doc

  kaixih

  committed
  5242c7a

  Copy full SHA for 5242c7a

Commits on Apr 7, 2025

• Address comments

  kaixih

  committed
  1fd1937

  Copy full SHA for 1fd1937

Commits on Apr 4, 2025

• Support fp8 einsum

  kaixih

  committed
  e0e253b

  Copy full SHA for e0e253b

• Merge pull request #4681 from IvyZX:lora

  Show description for e58aeea
  

  Flax Authors

  committed
  e58aeea

  Copy full SHA for e58aeea

• Make LoRA's dtype arg useful

  IvyZX

  committed
  0ca0271

  Copy full SHA for 0ca0271

Commits on Apr 3, 2025

• fix metadata sharing in nn.Variable.replace

  Show description for e16cfc4
  

  liudangyi

  authored and

  Flax Authors

  committed
  e16cfc4

  Copy full SHA for e16cfc4

• Automatic model selection for Gemma 3 models.

  Show description for 031d745
  

  casaro

  authored and

  Flax Authors

  committed
  031d745

  Copy full SHA for 031d745

• Add support for layer-specific rope scale factors.

  Show description for 97ed31e
  

  casaro

  authored and

  Flax Authors

  committed
  97ed31e

  Copy full SHA for 97ed31e

• Sow top activations based on absolute value.

  Show description for 5d41f6a
  

  casaro

  authored and

  Flax Authors

  committed
  5d41f6a

  Copy full SHA for 5d41f6a

Commits on Mar 31, 2025

• Merge pull request #4656 from google:v0.10.5

  Show description for 68b69f7
  

  Flax Authors

  committed
  68b69f7

  Copy full SHA for 68b69f7

• 0.10.5

  cgarciae

  committed
  01254af

  Copy full SHA for 01254af

Commits on Mar 28, 2025

• deflake a test

  Show description for 6abddc9
  

  levskaya

  authored and

  Flax Authors

  committed
  6abddc9

  Copy full SHA for 6abddc9

Commits on Mar 27, 2025

• Merge pull request #4657 from IvyZX:wrapper-typo

  Show description for 7600ad1
  

  Flax Authors

  committed
  7600ad1

  Copy full SHA for 7600ad1

• Merge pull request #4659 from IvyZX:IvyZX-patch-1

  Show description for d6ebb9d
  

  Flax Authors

  committed
  d6ebb9d

  Copy full SHA for d6ebb9d

• ubuntu upgrade 24.04

  IvyZX

  authored
  983a088

  Copy full SHA for 983a088

Commits on Mar 26, 2025

• Fix lazy_init typo

  IvyZX

  authored
  ce6e249

  Copy full SHA for ce6e249

• Fix flaky vmap test tolerance.

  Show description for 0dec8a8
  

  levskaya

  authored and

  Flax Authors

  committed
  0dec8a8

  Copy full SHA for 0dec8a8

Commits on Mar 25, 2025

• [nnx] remove transforms.deprecated

  Show description for ce48e5d
  cgarciae

  committed
  ce48e5d

  Copy full SHA for ce48e5d

• Merge pull request #4649 from IvyZX:attn

  Show description for 527cb4a
  

  Flax Authors

  committed
  527cb4a

  Copy full SHA for 527cb4a

• Use jax.nn.dot_product_attention when possible

  IvyZX

  committed
  6f7322d

  Copy full SHA for 6f7322d

• Merge pull request #4648 from IvyZX:fix-head

  Show description for 5f7af86
  

  Flax Authors

  committed
  5f7af86

  Copy full SHA for 5f7af86

Pagination

Previous

Next