google/flax

Pull Requests Commits

Merge pull request #5526 from samanklesaria:fix_ci_precision

a-googler committed 5 days ago

a31129e2

Use less precise numbers in linen doctest

samanklesaria committed 6 days ago

e2d7ba2d

Automated Code Change

h-joo committed 10 days ago

18f8f311

Merge pull request #5475 from dparikh79:fix/5461-nnx-clone-shares-buffers

a-googler committed 13 days ago

b196b016

nnx.clone: add arrays parameter to also copy underlying buffers

dparikh79 committed 13 days ago

b1b1c184

Support dynamic attributes for Pyrefly in classes using HAS_DYNAMIC_ATTRIBUTES

QEDady committed 14 days ago

07560a7a

Merge pull request #5480 from google:docs-examples--od

a-googler committed 16 days ago

29e55c36

Merge pull request #5505 from mohsinm-dev:fix-attention-assert-to-valueerror

a-googler committed 16 days ago

51b3d8f1

Merge pull request #5504 from mohsinm-dev:fix-nnx-promote-dtype-docstring

a-googler committed 16 days ago

067a571d

Merge pull request #5479 from google:fix-typos-pth-to-nnx-migration

a-googler committed 16 days ago

dd946850

Merge pull request #5511 from google:vfdev-5-patch-1

a-googler committed 20 days ago

dcfabd05

Update tensorflow dependency versions in pyproject.toml

vfdev-5 committed 20 days ago

Verified 395e0331

update to v0.12.8

Cristian Garcia committed 23 days ago

3a491108

optimize jit_partial

Cristian Garcia committed 25 days ago

0a0aca74

Fixed typos in docs_nnx/migrating/pytorch_to_jax_flax.rst

vfdev-5 committed 25 days ago

b070daad

replace bare asserts with ValueError in attention validation

mohsinm-dev committed 25 days ago

4f85c88e

Added Object Detection example with minimal DETR

vfdev-5 committed 25 days ago

25785b10

fix nnx promote_dtype docstring

mohsinm-dev committed 25 days ago

1f3d9549

Merge pull request #5507 from google:add-dependabot

a-googler committed 25 days ago

0682fe4d

Merge pull request #5506 from google:fix-ci-nest-asyncio

a-googler committed 25 days ago

6fec30fa

vfdev-5 committed 25 days ago

5768bb59

Add Dependabot configuration for dependency updates

vfdev-5 committed 25 days ago

Verified 0a682323

Merge pull request #5455 from google:docs-examples--vit

a-googler committed 26 days ago

e48ce367

Merge pull request #5492 from mohsinm-dev:fix-multimetric-hashable-graphdef

a-googler committed 26 days ago

4c59d8f3

fix MultiMetric GraphDef not hashable under jit (#5489)

mohsinm-dev committed 32 days ago

Verified 6f745922

Merge pull request #5477 from google:fix-gemma-sampler-test-kaggle

a-googler committed 37 days ago

5ab9d146

Merge branch 'main' into docs-examples--vit

vfdev-5 committed 37 days ago

Verified 4ad282fb

Merge pull request #5463 from google:docs-examples--unetr

a-googler committed 40 days ago

77a603be

Added image segmentation example

vfdev-5 committed 40 days ago

094b1a13

Added ViT training example

vfdev-5 committed 40 days ago

a450307f

Older