google/flax

Pull Requests Commits
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
Release 0.6.11
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
Terms Of Use Privacy Policy Refund Policy Impressum