google/flax

Pull Requests Commits

Internal change.

pschuh committed 2 years ago

1db9adaa

Merge pull request #3161 from chiamp:fix_broken_links

a-googler committed 2 years ago

d4d137c2

fixed broken links

chiamp committed 2 years ago

7a644f71

Merge pull request #3162 from chiamp:relax_tol

a-googler committed 2 years ago

25ab83c3

relaxed tolerance

chiamp committed 2 years ago

3e470bff

Merge pull request #3158 from google:haiku-migration-guide-transforms

a-googler committed 2 years ago

115b8a5e

Merge pull request #3133 from google:update-to-gymnasium

a-googler committed 2 years ago

f447e631

Merge pull request #3160 from levskaya:fix_doc_reqs

a-googler committed 2 years ago

50684adb

update doctest requirements to use more recent jax minver

levskaya committed 2 years ago

abfd7e73

Merge pull request #3073 from google:improve-toc

a-googler committed 2 years ago

11277dc6

Remove flax_relaxed_naming feature flag.

levskaya committed 2 years ago

befa4c1b

add sections to flax.linen toctree

cgarciae committed 2 years ago

ca867572

replace gym with gymnasium

cgarciae committed 2 years ago

2ec8d90c

delete long-unused dotgetter utility

levskaya committed 2 years ago

4f9f64e8

add lifted transforms to haiku migration guide

cgarciae committed 2 years ago

d5a25b77

Merge pull request #3155 from IvyZX:push

a-googler committed 2 years ago

fd567c85

IvyZX committed 2 years ago

6ad44b73

Merge pull request #3016 from IvyZX:jit-guide

a-googler committed 2 years ago

bdb68ef5

Merge pull request #3153 from levskaya:trafo_setup_fix

a-googler committed 2 years ago

875b40fe

Merge pull request #3154 from LenaMartens:haiku

a-googler committed 2 years ago

d197ca1a

Apply suggestions from code review

levskaya committed 2 years ago

Verified 98395e99

Update Haiku migration guide.

LenaMartens committed 2 years ago

038d512e

Fix setup trigger issue under sharing and transforms.

levskaya committed 2 years ago

5a82a055

Merge pull request #3139 from google:haiku-migration-guide

a-googler committed 2 years ago

fcd914d8

add haiku_migration_guide

cgarciae committed 2 years ago

ea474fa0

Update Scale up Flax Modules on multiple devices

8bitmp3 committed 2 years ago

87e7fa75

upgrade jit guide

IvyZX committed 2 years ago

acce3a28

Fix Orbax aggregate criteria and bump back to latest Orbax version

IvyZX committed 2 years ago

00a028a4

Merge pull request #3077 from google:eager-bind-submodules

a-googler committed 2 years ago

7b7b8bb5

Merge pull request #3122 from google:levskaya-prelu-1

a-googler committed 2 years ago

e728685c

Older