google/flax

Pull Requests Commits

Merge pull request #3172 from chiamp:add_absl

a-googler committed 2 years ago

b05c6738

added absltest to linen_recurrent_test

chiamp committed 2 years ago

6bdb1c36

Merge pull request #3167 from chiamp:fix_reference_link

a-googler committed 2 years ago

517225e6

fixed incorrect reference link

chiamp committed 2 years ago

febceabb

Merge pull request #3170 from chiamp:fix_precommit

a-googler committed 2 years ago

3efcd8a2

fix precommit issues

chiamp committed 2 years ago

2d02cc79

Flax is changing the `RNNCellBase` API:

chiamp committed 2 years ago

b6ab7d69

Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change.

cpgaffney1 committed 2 years ago

842a2874

Merge pull request #3169 from google:haiku-migration-guide-fixes

a-googler committed 2 years ago

7c9c67f8

Merge pull request #3168 from google:update-python-version-support

a-googler committed 2 years ago

bc122df9

improve haiku migration guide

cgarciae committed 2 years ago

b33f960e

Merge pull request #3046 from canyon289:update_main

a-googler committed 2 years ago

fe93a7ef

update python version support

cgarciae committed 2 years ago

7b481c4c

Merge pull request #3109 from chiamp:dict_migration_guide

a-googler committed 2 years ago

01eb19cc

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

Newer Older