google/flax

Pull Requests Commits

add __init__ note

cgarciae committed 2 years ago

d8bd44ce

cgarciae committed 2 years ago

b6042886

cgarciae committed 2 years ago

dfcf4c9f

cgarciae committed 2 years ago

282d1e57

On top of #3217, make sure the shape compatibility works before the next Orbax release.

IvyZX committed 2 years ago

3741278d

Merge pull request #3231 from google:fix-scan-docs

a-googler committed 2 years ago

f1032336

improve scan docs

cgarciae committed 2 years ago

93d48c19

Let Flax-Orbax to not port the shape of `target` arrays when they port the `target` shardings.

IvyZX committed 2 years ago

719217b6

Merge pull request #3223 from IvyZX:orbax

a-googler committed 2 years ago

2497f823

Fix checkpointing guide

IvyZX committed 2 years ago

2857f30f

Merge pull request #3195 from google:haiku-migration-scan-over-layers

a-googler committed 2 years ago

9ba7cc0c

Merge pull request #3219 from google:fix-main

a-googler committed 2 years ago

4f1884c7

fix update_requirements.py

cgarciae committed 2 years ago

c90c2670

add preview to pyink

cgarciae committed 2 years ago

f86fd1c4

cgarciae committed 2 years ago

0f4e6623

Merge pull request #3216 from google:use-pyink

a-googler committed 2 years ago

3ab11fe2

fix mypy + add .git-blame-ignore-revs

cgarciae committed 2 years ago

97d038cc

cgarciae committed 2 years ago

40a6e074

cgarciae committed 2 years ago

338caabf

Merge pull request #3197 from google:tabulate-wrap-cell

a-googler committed 2 years ago

39e3e326

Refactor TypeHandler to operate over batches of values, rather than individual ones. This allows more flexibility for implementations that may operate more efficiently on batches.

cpgaffney1 committed 2 years ago

c8bb930f

[linen] More minor cleanup in normalization `compute_stats`.

chr1sj0nes committed 2 years ago

b4591c14

Migrating Flax from returning FrozenDicts to returning regular dicts. More details can be found in this [announcement](https://github.com/google/flax/discussions/3191)

chiamp committed 2 years ago

aa7b14e1

Merge pull request #3215 from chiamp:utility_fn

a-googler committed 2 years ago

dfb4fc34

Merge pull request #3209 from marcoselvi:patch-1

a-googler committed 2 years ago

ac6cde09

updated flax guides to use utility fns

chiamp committed 2 years ago

a0dbe390

Merge pull request #3214 from chiamp:fix_flax_basics

a-googler committed 2 years ago

54a0296b

make flax_basics guide use utility fns

chiamp committed 2 years ago

93a38ffc

Merge pull request #3210 from chiamp:dict_migration

a-googler committed 2 years ago

290e50f8

Merge pull request #3213 from google:fix-rnn-issue

a-googler committed 2 years ago

b234192c

Older