Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jul 28, 2022

• Merge pull request #2343 from Chuxiaof:patch-1

  Show description for 67d7435
  

  Flax Authors

  committed
  67d7435

  Copy full SHA for 67d7435

• Merge pull request #2277 from cgarciae:split-ci

  Show description for 29e385a
  

  Flax Authors

  committed
  29e385a

  Copy full SHA for 29e385a

• Merge pull request #2291 from cgarciae:improve-flax-basics

  Show description for 33a2fc1
  

  Flax Authors

  committed
  33a2fc1

  Copy full SHA for 33a2fc1

• Merge pull request #2317 from cgarciae:document-repo-analytics

  Show description for 2ffc59e
  

  Flax Authors

  committed
  2ffc59e

  Copy full SHA for 2ffc59e

• Merge branch 'main' into improve-flax-basics

  cgarciae

  committed
  1cc7603

  Copy full SHA for 1cc7603

• Merge pull request #2339 from sanchit-gandhi:add-unroll-to-scan-with-axes

  Show description for 6ae2268
  

  Flax Authors

  committed
  6ae2268

  Copy full SHA for 6ae2268

• Further fix the singular-leaf checkpointing, and add tests.

  Show description for 035693c
  

  IvyZX

  authored and

  Flax Authors

  committed
  035693c

  Copy full SHA for 035693c

Commits on Jul 27, 2022

• Fix type annotation for step; close #2342

  Chuxiaof

  authored
  eafc096

  Copy full SHA for eafc096

• Merge pull request #2325 from levskaya:treesymbolfix

  Show description for 0740ef6
  

  Flax Authors

  committed
  0740ef6

  Copy full SHA for 0740ef6

• Forward unroll argument in scan_with_axes

  

  sanchit-gandhi

  committed
  a9fdc47

  Copy full SHA for a9fdc47

• Merge pull request #2338 from levskaya:doclinks

  Show description for 8d3e987
  

  Flax Authors

  committed
  8d3e987

  Copy full SHA for 8d3e987

• Replace all jax.tree_* calls with jax.tree_util.tree_*

  Show description for 4c2d87b
  levskaya

  committed
  4c2d87b

  Copy full SHA for 4c2d87b

• 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

• use absl instead of typer

  cgarciae

  committed
  37fea7b

  Copy full SHA for 37fea7b

• 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

• add all requirements

  cgarciae

  committed
  c17f377

  Copy full SHA for c17f377

• clean up + document repo analytics

  cgarciae

  committed
  55b94fd

  Copy full SHA for 55b94fd

• 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