Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jan 4, 2024

• Merge pull request #3600 from google:nnx-clean-flaglib

  Show description for cb6843f
  

  Flax Authors

  committed
  cb6843f

  Copy full SHA for cb6843f

• Merge pull request #3581 from google:nnx-submodule-iterator

  Show description for 3b23351
  

  Flax Authors

  committed
  3b23351

  Copy full SHA for 3b23351

Commits on Jan 3, 2024

• delete flaglib duplicated copyright comment

  cgarciae

  committed
  039845c

  Copy full SHA for 039845c

• [nnx] add submodule iterator

  cgarciae

  committed
  1e75509

  Copy full SHA for 1e75509

• Merge pull request #3591 from shoyer:patch-1

  Show description for 85eb8c0
  

  Flax Authors

  committed
  85eb8c0

  Copy full SHA for 85eb8c0

• Merge pull request #3582 from google:nnx-flagslib-add-get-overloads

  Show description for cff22d3
  

  Flax Authors

  committed
  cff22d3

  Copy full SHA for cff22d3

• Merge pull request #3583 from google:nnx-add-missing-symbols

  Show description for bfe04e5
  

  Flax Authors

  committed
  bfe04e5

  Copy full SHA for bfe04e5

• Merge pull request #3596 from google:nnx-why-open-in-colab

  Show description for b28c0f7
  

  Flax Authors

  committed
  b28c0f7

  Copy full SHA for b28c0f7

• Merge pull request #3578 from google:nnx-fix-variable-unflatten

  Show description for 2d15e85
  

  Flax Authors

  committed
  2d15e85

  Copy full SHA for 2d15e85

• Merge pull request #3592 from google:nnx-improve-tuple-set

  Show description for 4fb5604
  

  Flax Authors

  committed
  4fb5604

  Copy full SHA for 4fb5604

• Merge pull request #3580 from google:nnx-get-named-sharding

  Show description for e8a14af
  

  Flax Authors

  committed
  e8a14af

  Copy full SHA for e8a14af

• Merge pull request #3587 from chiamp:error_format

  Show description for b48d05f
  

  Flax Authors

  committed
  b48d05f

  Copy full SHA for b48d05f

Commits on Jan 2, 2024

• add open in colab button

  cgarciae

  committed
  b6a096b

  Copy full SHA for b6a096b

Commits on Jan 1, 2024

• improve graph_utils._set_key_tuple

  cgarciae

  committed
  3fcc013

  Copy full SHA for 3fcc013

Commits on Dec 27, 2023

• Fix type in NNX readme

  shoyer

  authored
  1c7ff8b

  Copy full SHA for 1c7ff8b

Commits on Dec 21, 2023

• fixed error doc formatting

  chiamp

  committed
  62859a6

  Copy full SHA for 62859a6

• Merge pull request #3571 from giovannic:bug/attention_promotion

  Show description for 85dfad2
  

  Flax Authors

  committed
  85dfad2

  Copy full SHA for 85dfad2

• [nnx] export missing symbols

  cgarciae

  committed
  03ffeac

  Copy full SHA for 03ffeac

• [nnx] flaglib add get overloads

  cgarciae

  committed
  2eebcc0

  Copy full SHA for 2eebcc0

• [nnx] improve spmd

  cgarciae

  committed
  3b77280

  Copy full SHA for 3b77280

• nnx: fix variable unflatten

  cgarciae

  committed
  0bf48f6

  Copy full SHA for 0bf48f6

Commits on Dec 20, 2023

• Merge pull request #3574 from chiamp:nnx_initializers

  Show description for 008c3cc
  

  Flax Authors

  committed
  008c3cc

  Copy full SHA for 008c3cc

Commits on Dec 19, 2023

• Merge pull request #3573 from google:nnx-improve-readme

  Show description for f1cb011
  

  Flax Authors

  committed
  f1cb011

  Copy full SHA for f1cb011

• expose ones and zeros initializers

  chiamp

  committed
  b5ec12b

  Copy full SHA for b5ec12b

• Merge pull request #3567 from google:nnx-add-dropout-test

  Show description for 3e56f8e
  

  Flax Authors

  committed
  3e56f8e

  Copy full SHA for 3e56f8e

• Merge pull request #3555 from chiamp:nnx_initializers

  Show description for dad384c
  

  Flax Authors

  committed
  dad384c

  Copy full SHA for dad384c

• Merge pull request #3527 from NeilGirdhar:key_type_error

  Show description for 1f5947f
  

  Flax Authors

  committed
  1f5947f

  Copy full SHA for 1f5947f

• nnx: add interacting with JAX section to README

  cgarciae

  committed
  0a67fc7

  Copy full SHA for 0a67fc7

Commits on Dec 18, 2023

• Fix promotion bug in MultiHeadDotProductAttention:

  Show description for 1c27546
  giovannic

  committed
  1c27546

  Copy full SHA for 1c27546

• Merge pull request #3561 from chiamp:typo

  Show description for f154a64
  

  Flax Authors

  committed
  f154a64

  Copy full SHA for f154a64

• Merge pull request #3563 from chiamp:readme

  Show description for ca7d748
  

  Flax Authors

  committed
  ca7d748

  Copy full SHA for ca7d748

• Merge pull request #3566 from google:nnx-attention-add-qkv-features

  Show description for a59675c
  

  Flax Authors

  committed
  a59675c

  Copy full SHA for a59675c

• Merge pull request #3547 from google:nnx-rngs-impl-getattr

  Show description for eb1bfb1
  

  Flax Authors

  committed
  eb1bfb1

  Copy full SHA for eb1bfb1

• Passed previously unused `iterate_clz` to `register_pytree_with_keys`.

  Show description for 0697987
  

  superbobry

  authored and

  Flax Authors

  committed
  0697987

  Copy full SHA for 0697987

• Merge pull request #3565 from google:cgarciae-nnx-fix-readme-install

  Show description for fa2d80e
  

  Flax Authors

  committed
  fa2d80e

  Copy full SHA for fa2d80e