Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Aug 5, 2024

• Merge pull request #4056 from cool-RR:patch-2

  Show description for 5694517
  

  Flax Authors

  committed
  5694517

  Copy full SHA for 5694517

Commits on Aug 1, 2024

• Merge pull request #4102 from google:linen-transparent

  Show description for d20f594
  

  Flax Authors

  committed
  d20f594

  Copy full SHA for d20f594

Commits on Jul 31, 2024

• Customizing the einsums allows us to use the quantized einsum layers provided by [AQT](https://github.com/google/aqt).

  Show description for cd6218f
  

  Flax Team

  committed
  cd6218f

  Copy full SHA for cd6218f

Commits on Jul 30, 2024

• set profile_duration_ms = None as in periodic_actions there's default value for both num_profile_steps and profile_duration_ms, and the profile stopping condition is when both num_profile_steps and…

  Show description for 5b12e9b
  

  ZhaoyueCheng

  authored and

  Flax Authors

  committed
  5b12e9b

  Copy full SHA for 5b12e9b

Commits on Jul 26, 2024

• [numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

  Show description for 5172c78
  

  hawkinsp

  authored and

  Flax Authors

  committed
  5172c78

  Copy full SHA for 5172c78

Commits on Jul 25, 2024

• [linen] add share_scope

  cgarciae

  committed
  36952ea

  Copy full SHA for 36952ea

Commits on Jul 23, 2024

• fix restore_checkpoint example in docstring

  Show description for 46dc51e
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  46dc51e

  Copy full SHA for 46dc51e

• Merge pull request #4079 from danielwatson6:scope-shape-check

  Show description for d718655
  

  Flax Authors

  committed
  d718655

  Copy full SHA for d718655

Commits on Jul 22, 2024

• Merge pull request #4084 from google:nnx-split-merge-inputs

  Show description for 8a88081
  

  Flax Authors

  committed
  8a88081

  Copy full SHA for 8a88081

• Merge pull request #4099 from google:nnx-reseed

  Show description for a7bdadb
  

  Flax Authors

  committed
  a7bdadb

  Copy full SHA for a7bdadb

• Merge pull request #4081 from IvyZX:bridge

  Show description for d8bc194
  

  Flax Authors

  committed
  d8bc194

  Copy full SHA for d8bc194

• Add shape-based lazy init to LinenToNNX (prev LinenWrapper)

  IvyZX

  committed
  0139c90

  Copy full SHA for 0139c90

• [nnx] add reseed

  cgarciae

  committed
  3f8c62f

  Copy full SHA for 3f8c62f

• [nnx] add split/merge_inputs

  cgarciae

  committed
  8299312

  Copy full SHA for 8299312

• [linen] fold rngs on jit to improve caching

  Show description for afcaf66
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  afcaf66

  Copy full SHA for afcaf66

• [linen] generalize transform caching

  Show description for 4e83e09
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  4e83e09

  Copy full SHA for 4e83e09

• Merge pull request #4098 from google:fix-main

  Show description for 65cd19e
  

  Flax Authors

  committed
  65cd19e

  Copy full SHA for 65cd19e

• fix mypy failures due tu numpy update

  cgarciae

  committed
  30b0b0a

  Copy full SHA for 30b0b0a

Commits on Jul 18, 2024

• Merge pull request #4092 from IvyZX:unbreak

  Show description for ceacc09
  

  Flax Authors

  committed
  ceacc09

  Copy full SHA for ceacc09

• Ignore Orbax warning in deprecated flax.training and unbreak head

  IvyZX

  committed
  b0ebefc

  Copy full SHA for b0ebefc

Commits on Jul 17, 2024

• Merge pull request #4004 from google:nnx-pure

  Show description for c4066cc
  

  Flax Authors

  committed
  c4066cc

  Copy full SHA for c4066cc

Commits on Jul 16, 2024

• Merge pull request #4059 from google:nnx-improve-filters-guide

  Show description for 62127f8
  

  Flax Authors

  committed
  62127f8

  Copy full SHA for 62127f8

• [nnx] improve filters guide

  cgarciae

  committed
  7937ca0

  Copy full SHA for 7937ca0

• Merge pull request #4077 from knightXun:patch-1

  Show description for 93babfd
  

  Flax Authors

  committed
  93babfd

  Copy full SHA for 93babfd

Commits on Jul 15, 2024

• Merge pull request #4078 from google:nnx-extract-apis

  Show description for e7f43dd
  

  Flax Authors

  committed
  e7f43dd

  Copy full SHA for e7f43dd

• [nnx] add extract APIs

  cgarciae

  committed
  16c2c32

  Copy full SHA for 16c2c32

Commits on Jul 12, 2024

• Merge pull request #4076 from kaixih:add_fp8_doc

  Show description for e8021e9
  

  Flax Authors

  committed
  e8021e9

  Copy full SHA for e8021e9

• Perform shape checks for self.param AFTER unboxing, make shmap users happy

  danielwatson6

  committed
  bc68908

  Copy full SHA for bc68908

• [example]: remove lm1b useless parallism rules

  knightXun

  authored
  9329127

  Copy full SHA for 9329127

Commits on Jul 11, 2024

• Add doc for fp8

  kaixih

  committed
  350d9f8

  Copy full SHA for 350d9f8

• Reverting unsafe_pytree #4030

  Show description for 239d4e6
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  239d4e6

  Copy full SHA for 239d4e6

• [nnx] pure API

  cgarciae

  committed
  f89e62c

  Copy full SHA for f89e62c

• Remove some outdated backward-compatibility code.

  Show description for 5c97143
  

  IvyZX

  authored and

  Flax Authors

  committed
  5c97143

  Copy full SHA for 5c97143

Commits on Jul 10, 2024

• Modify Flax checkpointing in preparation for cl/650338576.

  Show description for e59b14e
  

  cpgaffney1

  authored and

  Flax Authors

  committed
  e59b14e

  Copy full SHA for e59b14e

• Merge pull request #4070 from google:nnx-improve-mnist-tutorial

  Show description for 1b58348
  

  Flax Authors

  committed
  1b58348

  Copy full SHA for 1b58348