google/flax

Pull Requests Commits

Deduplicate contributing and philosophy and move to main site

IvyZX committed 217 days ago

103f65de

Preserve sharding information in axes_scan

oliverdutton-iso committed 218 days ago

4e2becd8

Internal change

kho committed 218 days ago

cbb4f80f

Merge pull request #4808 from google:to-linen-method-calls

a-googler committed 218 days ago

171b28e8

[nnx] allow method calls in ToLinen

cgarciae committed 218 days ago

61d0e9e0

[nnx] remove VariableState

cgarciae committed 221 days ago

e8984b70

Merge pull request #4755 from google:nnx-remove-all

a-googler committed 225 days ago

99f8e598

Merge pull request #4798 from google:ver-0.10.7

a-googler committed 225 days ago

1ecd34f9

update version to 0.10.7

cgarciae committed 225 days ago

d2cf8c2b

Fix typo in Flax `nnx_basics` doc.

a-googler committed 231 days ago

cca78723

Internal changes

basioli-k committed 231 days ago

6770f73f

Merge pull request #4775 from google:try-fix-colab-link-nnx

a-googler committed 232 days ago

7d22696c

Fixed nnx.remat docstring rendering

vfdev-5 committed 233 days ago

Verified 4870afe5

Merge pull request #4788 from google:add

a-googler committed 233 days ago

0b9bb4ee

Exposed OptState in nnx module

vfdev-5 committed 233 days ago

755067df

Use `.input_formats` and `.output_formats` in place of `.input_layouts` and `.output_layouts` respectively.

yashk2810 committed 238 days ago

dcd2f5e2

[nnx] mutable array p3

cgarciae committed 241 days ago

571c9e40

[nnx] add optax optimizer

cgarciae committed 244 days ago

0a4e8ef1

Merge pull request #4766 from IvyZX:bdg

a-googler committed 244 days ago

21e64ece

Simplify ToNNX access of Linen module methods

IvyZX committed 244 days ago

6a6561e5

vfdev-5 committed 247 days ago

Verified 62238aaf

Added a note to gemma guide about model's license consent on kaggle

vfdev-5 committed 247 days ago

Verified 3945f066

Try to fix colab link for nnx docs

vfdev-5 committed 247 days ago

Verified 63a53ae4

Modified CNN model to have batch norm

sanepunk committed 250 days ago

11d3d2c7

revise axes_scan to flatten argument pytrees only once

mattjj committed 251 days ago

893a6602

Merge pull request #4770 from IvyZX:fix-links

a-googler committed 252 days ago

8233415b

Fix bad doc links

IvyZX committed 252 days ago

2d4df43e

Merge pull request #4768 from hamogu:patch-1

a-googler committed 252 days ago

c5d5ab1e

Merge pull request #4767 from nireekshak:test-branch

a-googler committed 253 days ago

2a9e5f2d

Merge pull request #4563 from muhrin:fix-1925-linen-nonstring-keys

a-googler committed 253 days ago

f525c029

Newer Older