google/flax

Pull Requests Commits

test if internal formatter formats on save, and if so, whether CI will break

chiamp committed 2 years ago

05a1bc2d

Merge pull request #3463 from levskaya:vjp_fix

a-googler committed 2 years ago

e2c3dfd3

Add simple argument-only lifted nn.grad and nn.value_and_grad functions.

levskaya committed 2 years ago

5ff36ba4

Merge pull request #3461 from levskaya:copymethod

a-googler committed 2 years ago

85245ada

Add user-friendly module copy method.

levskaya committed 2 years ago

40e2b931

Merge pull request #3459 from levskaya:clonefix

a-googler committed 2 years ago

055e28fa

Fix Module.clone in deepclone mode for internal usage.

levskaya committed 2 years ago

5587371b

Merge pull request #3448 from superbobry:fork-configuration

a-googler committed 2 years ago

c1023d90

Forked a subset of JAX configuration APIs

superbobry committed 2 years ago

44c2a886

Merge pull request #3434 from chiamp:docs

a-googler committed 2 years ago

0b126b83

restructured doc folders

chiamp committed 2 years ago

1383ea30

Merge pull request #3455 from chiamp:format

a-googler committed 2 years ago

1f71cdc1

added pre-commit hook that sort imports and formats

chiamp committed 2 years ago

6fa38399

Merge pull request #3441 from kaixih:update_scale_compute

a-googler committed 2 years ago

8d097722

Merge pull request #3451 from chiamp:quickstart

a-googler committed 2 years ago

6f69bb6a

fixed quickstart

chiamp committed 2 years ago

a96626ac

Add merge / finalize step when using OCDBT driver. Files will be first written to per-process subdirectories, which are later copied by reference to the main directory before the checkpoint is finalized.

cpgaffney1 committed 2 years ago

c0087535

Merge pull request #3446 from levskaya:rc7.6

a-googler committed 2 years ago

4c458b1e

bump version number

levskaya committed 2 years ago

3b0b5486

Merge pull request #3444 from levskaya:rc7.5

a-googler committed 2 years ago

791e7da0

update for release v0.7.5

levskaya committed 2 years ago

332e7236

Merge pull request #3443 from 8bitmp3:basics_visible

a-googler committed 2 years ago

9d6e7d83

Make Flax Basics visible

8bitmp3 committed 2 years ago

Verified fc4e43fa

Merge pull request #3418 from chiamp:mgu

a-googler committed 2 years ago

38894a1d

added MGU class

chiamp committed 2 years ago

f70174f5

Merge pull request #3302 from chiamp:lm1b

a-googler committed 2 years ago

8db32aed

Merge pull request #3425 from chiamp:rnn_warning

a-googler committed 2 years ago

416b3e20

Merge pull request #3385 from chiamp:early_stopping

a-googler committed 2 years ago

36bbb414

added has_improved field to EarlyStopping

chiamp committed 2 years ago

b620117b

Merge pull request #3440 from chiamp:fix_fp8

a-googler committed 2 years ago

90641085

Older