google/flax

Pull Requests Commits

Merge pull request #2831 from cgarciae:improve-map_variables-v2

a-googler committed 3 years ago

f2094bac

Merge pull request #2769 from cgarciae:fix-tabuluate-non-array-values

a-googler committed 3 years ago

8c428faf

Update flax/linen/transforms.py

chiamp committed 3 years ago

Verified 5f67729d

improve the CasualDense example

cgarciae committed 3 years ago

93e5254e

import initializers module directly

chiamp committed 3 years ago

d696c706

Merge pull request #2818 from cgarciae:improve-map-variables-docs

a-googler committed 3 years ago

e54411d2

Merge pull request #2819 from cgarciae:cache-mypy

a-googler committed 3 years ago

cf27f3e8

Update jax dependency

jheek committed 3 years ago

c7fd31f4

Add clarification on mixing concrete and abstract vals

jheek committed 3 years ago

ed861304

Internal change

IvyZX committed 3 years ago

7ce79b3b

cache mypy cache on CI

cgarciae committed 3 years ago

8863549d

Merge pull request #2820 from IvyZX:push

a-googler committed 3 years ago

dc28d8cb

IvyZX committed 3 years ago

bc517e84

improve nn.map_variables docstrings

cgarciae committed 3 years ago

b97757cf

Implement lazy init

jheek committed 3 years ago

65ef2003

Merge pull request #2809 from cgarciae:string-methods

a-googler committed 3 years ago

a3092738

Merge pull request #2803 from cgarciae:repr-issue

a-googler committed 3 years ago

e83711be

Merge pull request #2767 from cgarciae:rename-conv-lstm

a-googler committed 3 years ago

a0a57ebf

Merge pull request #2578 from mathisgerdes:transpose-kernel

a-googler committed 3 years ago

6d85aac3

allow specifying method as string name in init

cgarciae committed 3 years ago

05053d96

Currently, by default, num_train_steps is 0, not -1.

nouiz committed 3 years ago

da775316

Doc a script requirement

nouiz committed 3 years ago

84ade75b

Merge pull request #2810 from nouiz:doc_error

a-googler committed 3 years ago

6bccee39

Fixing the punctuation

arcAman07 committed 3 years ago

Verified 3e133f5f

Fixing the punctuation

arcAman07 committed 3 years ago

Verified 34534cee

Add a comment about an error.

nouiz committed 3 years ago

c2972cf8

Merge pull request #2805 from chiamp:glossary_doc

a-googler committed 3 years ago

5d20e913

Merge pull request #2807 from chiamp:initializers_refactor

a-googler committed 3 years ago

0f7a8e2e

Merge pull request #2800 from 8bitmp3:guides-restructuring

a-googler committed 3 years ago

0dda7d41

Merge pull request #2807 from chiamp:initializers_refactor

a-googler committed 3 years ago

afd4c9f8

Newer Older