google/flax

Pull Requests Commits

Add commit-success file on GDA saves in GCS because GCS directory move is not atomic.

IvyZX committed 3 years ago

34726643

Use jax.named_scope for name stack rather than named_call.

levskaya committed 3 years ago

90c08ca5

Make importing tensorstore optional and move related type hints to comments.

IvyZX committed 3 years ago

8d855f92

Allow all processes to checkpoint when not using GDA

jheek committed 3 years ago

02f9f575

Add test for writing and restoring empty checkpoints.

agesmundo committed 3 years ago

d5830394

Fix traverse_util.unflatten_tree when the input is an empty dict.

IvyZX committed 3 years ago

1be36480

Merge pull request #2343 from Chuxiaof:patch-1

a-googler committed 3 years ago

67d7435b

Merge pull request #2277 from cgarciae:split-ci

a-googler committed 3 years ago

29e385a2

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

a-googler committed 3 years ago

33a2fc1e

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

a-googler committed 3 years ago

2ffc59ea

Merge branch 'main' into improve-flax-basics

cgarciae committed 3 years ago

1cc76032

Fix: anchor the link to the same link

Davidnet committed 3 years ago

c8023182

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

a-googler committed 3 years ago

6ae22681

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

IvyZX committed 3 years ago

035693ca

Fix type annotation for step; close #2342

Chuxiaof committed 3 years ago

Verified eafc0965

Merge pull request #2325 from levskaya:treesymbolfix

a-googler committed 3 years ago

0740ef63

Sync making a link a clickable link with the provided jupyter notebooks.

Davidnet committed 3 years ago

e47f4ee5

Make link a link on the getting started

Davidnet committed 3 years ago

Verified 952c57a1

Forward unroll argument in scan_with_axes

sanchit-gandhi committed 3 years ago

a9fdc47e

Merge pull request #2338 from levskaya:doclinks

a-googler committed 3 years ago

8d3e9870

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

levskaya committed 3 years ago

4c2d87ba

Fix stale URLs to read the docs site.

levskaya committed 3 years ago

8e88bce5

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

a-googler committed 3 years ago

2811ae51

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 success file when the commit is successful and check the existence of that file before deserialization.

yashk2810 committed 3 years ago

d6de016b

use absl instead of typer

cgarciae committed 3 years ago

37fea7bf

Merge pull request #2313 from levskaya:idfix

a-googler committed 3 years ago

db4a74b6

removed commented css directive

cgarciae committed 3 years ago

6ec1e574

Merge pull request #2330 from IvyZX:update-ver

a-googler committed 3 years ago

922e1eeb

add link to JAX installation page

cgarciae committed 3 years ago

34748358

ignore all _autosummary subfolders

cgarciae committed 3 years ago

c1126b0c

Newer Older