google/flax

Pull Requests Commits

cgarciae committed 2 years ago

e149d8c5

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

cgarciae committed 2 years ago

68b334e7

removed FrozenDict section

chiamp committed 2 years ago

f4386e41

fix carry slice logic

cgarciae committed 2 years ago

50f492b2

[linen] Use `stack` instead of `concatenate` in `compute_stats`, to handle scalar stats case.

chr1sj0nes committed 2 years ago

15d6857e

Fix typo in struct.py documentation

marcoselvi committed 2 years ago

Verified 0a4b0d68

Merge pull request #3207 from levskaya:fixgym

a-googler committed 2 years ago

881c4495

Fix warnings from atari gym.

levskaya committed 2 years ago

37e350e4

[JAX] Remove references to jax.config.jax_array.

hawkinsp committed 2 years ago

ce98f350

Fix norm calculation bug for 0-rank arrays.

levskaya committed 2 years ago

3e0d3aa3

Older