Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jul 27, 2022

• Merge pull request #2338 from levskaya:doclinks

  Show description for 8d3e987
  

  Flax Authors

  committed
  8d3e987

  Copy full SHA for 8d3e987

• Fix stale URLs to read the docs site.

  levskaya

  committed
  8e88bce

  Copy full SHA for 8e88bce

Commits on Jul 26, 2022

• Merge pull request #2177 from cgarciae:new-landing-page

  Show description for 2811ae5
  

  Flax Authors

  committed
  2811ae5

  Copy full SHA for 2811ae5

• Add `on_commit_callback` to put the responsibility of renaming the directories on the users of the serialization library. This will also fix the GCS atomic rename issue where the users can write a …

  Show description for d6de016
  

  yashk2810

  authored and

  Flax Authors

  committed
  d6de016

  Copy full SHA for d6de016

Commits on Jul 25, 2022

• Merge pull request #2313 from levskaya:idfix

  Show description for db4a74b
  

  Flax Authors

  committed
  db4a74b

  Copy full SHA for db4a74b

• removed commented css directive

  cgarciae

  committed
  6ec1e57

  Copy full SHA for 6ec1e57

• Merge pull request #2330 from IvyZX:update-ver

  Show description for 922e1ee
  

  Flax Authors

  committed
  922e1ee

  Copy full SHA for 922e1ee

• add link to JAX installation page

  cgarciae

  committed
  3474835

  Copy full SHA for 3474835

• ignore all _autosummary subfolders

  cgarciae

  committed
  c1126b0

  Copy full SHA for c1126b0

• redesign rdt page

  cgarciae

  committed
  b8833f2

  Copy full SHA for b8833f2

• 0.5.3 update version & changelog

  IvyZX

  committed
  95c568f

  Copy full SHA for 95c568f

Commits on Jul 23, 2022

• Fix checkpoints.py by checking and allowing single-leaf targets.

  Show description for 6e71622
  

  IvyZX

  authored and

  Flax Authors

  committed
  6e71622

  Copy full SHA for 6e71622

Commits on Jul 22, 2022

• Add GlobalDeviceArray/multihost checkpoint support to Flax.

  Show description for e4ef826
  

  IvyZX

  authored and

  Flax Authors

  committed
  e4ef826

  Copy full SHA for e4ef826

• Replace use of id() with global counter-based id.

  Show description for 963fa4a
  levskaya

  committed
  963fa4a

  Copy full SHA for 963fa4a

Commits on Jul 21, 2022

• cleanup: replace deprecated jax.tree_map with jax.tree_util.tree_map

  Show description for e7742de
  

  Qwlouse

  authored and

  Flax Authors

  committed
  e7742de

  Copy full SHA for e7742de

Commits on Jul 20, 2022

• Merge pull request #2310 from jheek:forward_axis_size-to-vmap

  Show description for 05d9095
  

  Flax Authors

  committed
  05d9095

  Copy full SHA for 05d9095

• Forward axis_size tot jax.vmap

  jheek

  committed
  cdc4a04

  Copy full SHA for cdc4a04

• Merge pull request #2306 from fuyw:fix_ppo_example_typo

  Show description for 80cb1b2
  

  Flax Authors

  committed
  80cb1b2

  Copy full SHA for 80cb1b2

Commits on Jul 19, 2022

• Update reference to tree_map to avoid deprecation warning.

  Show description for 4228004
  

  agesmundo

  authored and

  Flax Authors

  committed
  4228004

  Copy full SHA for 4228004

• fix ppo example typo

  fuyw

  committed
  78c38bf

  Copy full SHA for 78c38bf

Commits on Jul 18, 2022

• Fix nondeterministic bug arising from sharing logic during module adoption.

  Show description for 830f2ae
  

  levskaya

  authored and

  Flax Authors

  committed
  830f2ae

  Copy full SHA for 830f2ae

• Update reference to tree_map to avoid deprecation warning.

  Show description for da894ef
  

  agesmundo

  authored and

  Flax Authors

  committed
  da894ef

  Copy full SHA for da894ef

Commits on Jul 15, 2022

• Merge pull request #2296 from jheek:fix-var-aliasing

  Show description for 1ddedc0
  

  Flax Authors

  committed
  1ddedc0

  Copy full SHA for 1ddedc0

• Fix variable aliasing in put_variable

  jheek

  committed
  14e94c9

  Copy full SHA for 14e94c9

Commits on Jul 14, 2022

• Adds Reference to big_vision, t5x to "Repositories Using Flax".

  Show description for 9eddf86
  

  andsteing

  authored and

  Flax Authors

  committed
  9eddf86

  Copy full SHA for 9eddf86

Commits on Jul 9, 2022

• Fixes wrong type annotation on Module.__dir__. This causes problems when using multiple inheritance, since the subclass then uses the default return type rather than Module's one.

  Show description for e57133c
  

  marcvanzee

  authored and

  Flax Authors

  committed
  e57133c

  Copy full SHA for e57133c

Commits on Jul 8, 2022

• Merge pull request #2199 from YouJiacheng:patch-1

  Show description for 509bf97
  

  Flax Authors

  committed
  509bf97

  Copy full SHA for 509bf97

• Merge pull request #2281 from cgarciae:fix-type-annotation

  Show description for 05f5061
  

  Flax Authors

  committed
  05f5061

  Copy full SHA for 05f5061

• Merge pull request #2276 from cgarciae:fix-flax-basics

  Show description for 799312d
  

  Flax Authors

  committed
  799312d

  Copy full SHA for 799312d

• properly annotate int tuples

  cgarciae

  committed
  8d44178

  Copy full SHA for 8d44178

Commits on Jul 7, 2022

• Merge pull request #2276 from cgarciae:fix-flax-basics

  Show description for fabc7f0
  

  Flax Authors

  committed
  fabc7f0

  Copy full SHA for fabc7f0

• sync flax_basics notebook <--> markdown

  cgarciae

  committed
  4fab45d

  Copy full SHA for 4fab45d

• remove metadata

  cgarciae

  committed
  06c75f0

  Copy full SHA for 06c75f0

• add colab button to flax_basics

  cgarciae

  committed
  5fc5cbc

  Copy full SHA for 5fc5cbc

Commits on Jul 6, 2022

• Internal change.

  Show description for 81c3ab7
  

  Melissa Tan

  authored and

  Flax Authors

  committed
  81c3ab7

  Copy full SHA for 81c3ab7