google/flax

Pull Requests Commits

Fix broken links

a-googler committed 61 days ago

20237d63

Merge pull request #5491 from chenkuanliao:fix-issue-5483

a-googler committed 62 days ago

92cf9a7d

simplify vmapped metrics reset test

chenkuanliao committed 75 days ago

abc2dbe7

Merge pull request #5526 from samanklesaria:fix_ci_precision

a-googler committed 76 days ago

a31129e2

Use less precise numbers in linen doctest

samanklesaria committed 77 days ago

e2d7ba2d

Automated Code Change

h-joo committed 81 days ago

18f8f311

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

a-googler committed 84 days ago

b196b016

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

dparikh79 committed 84 days ago

b1b1c184

Support dynamic attributes for Pyrefly in classes using HAS_DYNAMIC_ATTRIBUTES

QEDady committed 85 days ago

07560a7a

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

a-googler committed 87 days ago

29e55c36

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

a-googler committed 87 days ago

51b3d8f1

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

a-googler committed 87 days ago

067a571d

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

a-googler committed 87 days ago

dd946850

Add vmap reset() regression test for nnx.metrics (#5483)

chenkuanliao committed 89 days ago

ca14973a

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

a-googler committed 91 days ago

dcfabd05

Update tensorflow dependency versions in pyproject.toml

vfdev-5 committed 91 days ago

Verified 395e0331

update to v0.12.8

Cristian Garcia committed 94 days ago

3a491108

optimize jit_partial

Cristian Garcia committed 96 days ago

0a0aca74

Fixed typos in docs_nnx/migrating/pytorch_to_jax_flax.rst

vfdev-5 committed 96 days ago

b070daad

replace bare asserts with ValueError in attention validation

mohsinm-dev committed 96 days ago

4f85c88e

Added Object Detection example with minimal DETR

vfdev-5 committed 96 days ago

25785b10

fix nnx promote_dtype docstring

mohsinm-dev committed 96 days ago

1f3d9549

Merge pull request #5507 from google:add-dependabot

a-googler committed 96 days ago

0682fe4d

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

a-googler committed 96 days ago

6fec30fa

vfdev-5 committed 96 days ago

5768bb59

Add Dependabot configuration for dependency updates

vfdev-5 committed 96 days ago

Verified 0a682323

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

a-googler committed 97 days ago

e48ce367

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

a-googler committed 97 days ago

4c59d8f3

fix MultiMetric GraphDef not hashable under jit (#5489)

mohsinm-dev committed 103 days ago

Verified 6f745922

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

a-googler committed 108 days ago

5ab9d146

Older