Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 1, 2024

• Merge pull request google#4229 from wenscarl:scale_type_fix

  Show description for b9bbc98
  

  Flax Authors

  committed
  b9bbc98

  Copy full SHA for b9bbc98

• fix scale dtype and refactor q_dot_dq

  wenscarl

  committed
  5edc65a

  Copy full SHA for 5edc65a

• [nnx] add flaxlib

  Show description for dfc86d5
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  dfc86d5

  Copy full SHA for dfc86d5

Commits on Sep 30, 2024

• Fix false positive tracer leaks in flax library.

  Show description for 62a6a25
  

  levskaya

  authored and

  Flax Authors

  committed
  62a6a25

  Copy full SHA for 62a6a25

Commits on Sep 26, 2024

• Merge pull request google#4205 from google:nnx-remove-registry

  Show description for 37f643d
  

  Flax Authors

  committed
  37f643d

  Copy full SHA for 37f643d

• Add pure dict conversion utils to .

  IvyZX

  committed
  3b9dce1

  Copy full SHA for 3b9dce1

• simplify graph.py

  cgarciae

  committed
  e7dba28

  Copy full SHA for e7dba28

• [nnx] add Randomness guide

  cgarciae

  committed
  65d4193

  Copy full SHA for 65d4193

Commits on Sep 23, 2024

• fix tests for numpy 2.0 compatibility

  Show description for fcd0071
  

  Cristian Garcia

  authored and

  Flax Authors

  committed
  fcd0071

  Copy full SHA for fcd0071

Commits on Sep 16, 2024

• Align bridge variable tree structures

  IvyZX

  committed
  4a04f75

  Copy full SHA for 4a04f75

• Partially revert google#4192 which sets back a bunch of previous merged pushes.

  Show description for 9eb0a61
  

  IvyZX

  authored and

  Flax Authors

  committed
  9eb0a61

  Copy full SHA for 9eb0a61

• split docs

  cgarciae

  committed
  ea57438

  Copy full SHA for ea57438

Commits on Sep 11, 2024

• Merge pull request google#4186 from IvyZX:errors

  Show description for b967964
  

  Flax Authors

  committed
  b967964

  Copy full SHA for b967964

• clean up axis hooks

  IvyZX

  committed
  978bc3d

  Copy full SHA for 978bc3d

Commits on Sep 10, 2024

• Merge nnx.errors to flax.errors

  IvyZX

  committed
  e8b0a1f

  Copy full SHA for e8b0a1f

• Merge pull request google#4161 from IvyZX:bdg-logic

  Show description for f948154
  

  Flax Authors

  committed
  f948154

  Copy full SHA for f948154

• Support linen.LogicallyPartitioned <-> nnx.Variable

  IvyZX

  committed
  a9cb80b

  Copy full SHA for a9cb80b

• [nnx] enable sharding transformation on integer prefixes

  cgarciae

  committed
  93c0627

  Copy full SHA for 93c0627

Commits on Sep 8, 2024

• [nnx] improve Optimizer metadata propagation

  cgarciae

  committed
  4c47287

  Copy full SHA for 4c47287

Commits on Sep 6, 2024

• Move all NNX content up a level to be equal with Linen, to make python packaging more consistent.

  Show description for 71b964b
  

  IvyZX

  authored and

  Flax Authors

  committed
  71b964b

  Copy full SHA for 71b964b

• Fixes a small bug in flax.linen.share_scope, where the scopes of children of the module being merged that were created before setup(),were not being updated to point to the new scope, and so they w…

  Show description for 05353e6
  

  marksandler2

  authored and

  Flax Authors

  committed
  05353e6

  Copy full SHA for 05353e6

Commits on Sep 5, 2024

• Internal change

  Show description for e848a99
  

  IvyZX

  authored and

  Flax Authors

  committed
  e848a99

  Copy full SHA for e848a99

Commits on Sep 4, 2024

• Merge pull request google#3922 from wenscarl:direct_quant

  Show description for c44b916
  

  Flax Authors

  committed
  c44b916

  Copy full SHA for c44b916

Commits on Sep 3, 2024

• Direct quantization for FP8 Dense Layer.

  wenscarl

  committed
  6f5cee1

  Copy full SHA for 6f5cee1

Commits on Aug 22, 2024

• use uv package manager

  cgarciae

  committed
  6197477

  Copy full SHA for 6197477

Commits on Aug 5, 2024

• Merge pull request google#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 google#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 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 22, 2024

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

Commits on Jul 10, 2024

• Allow metadata pass-through in flax.struct.field

  cool-RR

  committed
  f2f76d9

  Copy full SHA for f2f76d9

Commits on Jul 9, 2024

• uncomment some tests after 3.9 dropped

  IvyZX

  committed
  e800d5b

  Copy full SHA for e800d5b

Commits on Jul 2, 2024

• Rename the new dtype

  kaixih

  committed
  16f2866

  Copy full SHA for 16f2866

Pagination

Previous

Next