Go
Home
Pricing
FAQ
Install
Home
Pricing
FAQ
Install
Login
via GitHub
google/flax
Pull Requests
Commits
box-variable
Module-split-merge
add-cached-partial
add-migration-guides
add-mypy-plugin
andsteing-patch-1
array-refs-guide
avital-notes2
avital-patch-1
avital-patch-2
avital-patch-3
avital-patch-4
avital-patch-6
box-variable
bridge-improve-bind
compact_name_scope
cudnn-lstm
cursor-2
cython-test
deepseek-example
deepseek-example-v2
dependabot/cargo/flaxlib_src/cargo-f9635f71e3
dependabot/pip/examples/imagenet/pip-22842e8f01
dependabot/pip/examples/imagenet/pip-c711b36505
dependabot/pip/pip-5674335243
dependabot/uv/flaxlib_src/uv-4f61d8e847
deprecate-value
docs-build
docs-examples--od
docs-examples--vit
dynamic-rngs
errors
eval-shape-sharding
expose-module-functions
fix-bridge-boxes
fix-checkpointing-notebook
fix-ci-2
fix-dataset-path-examples
fix-doctest
fix-gemma
fix-jit
fix-lm1b
fix-mypy
fix-py314-kw-only-dataclasses
fix-site-banners
fix-summary
fix-to-linen
fix-typos-pth-to-nnx-migration
flax-nnx-page
flaxlib
functional-core
hello-ctx
hello-ctx-my-old-friend
hijax-in-graph
hijax-variable
improve-mnist-tutorial
improve-remat-docs
improve-scaling-up-guide
improve-scan
improve-scan-2
improve-tabulate-docs
inter-doc-test
keras-integration
keras-to-nnx
levskaya-patch-1
linen-fix-scan-out-axes
linen-fix-scan-out-axes-p2
linen-graph-node
linen-rtd
lsc-1743028162.4379792
main
marcvanzee-feature-request
marcvanzee-patch-1
marcvanzee-patch-2
marcvanzee-patch-3
marcvanzee-patch-4
marcvanzee-patch-5
marcvanzee-patch-6
marcvanzee-patch-7
marcvanzee-patch-8
marcvanzee-patch-9
marcvanzee-patch-10
marcvanzee-patch-11
marcvanzee-patch-12
nlp_seq_linen_mv
nlp_seq_linen_v2
nnx
nnx-add-call-to-basics
nnx-add-map-state
nnx-add-move
nnx-add-pmap
nnx-add-version
nnx-basics-tree
nnx-cache-flatten
nnx-custom-vjp-issue-full
nnx-debug-gc
nnx-demo
nnx-diffusion-example
nnx-empty-init
nnx-experimental
nnx-fast-flatten
nnx-fast-jit
nnx-filters-guide
nnx-fingerprint
nnx-fix-cond2
nnx-fix-jit
nnx-fix-sharing
nnx-flat-flatten
nnx-improve-filters
nnx-improve-mnist-tutorial
nnx-improve-nnx-basics
nnx-jit-new-update
nnx-leaves-are-leaves
nnx-lift-api
nnx-lm1b
nnx-mha-decode
nnx-mnist-tutorial-minor-changes
nnx-new-randomness-guide
nnx-optax-optimizer
nnx-optimizations-p4-rust
nnx-optimizations-p4
nnx-optimize-fingerprint
nnx-optimizer-use-variables
nnx-pytree-2
nnx-pytree-3
nnx-pytree-modules
nnx-pytree-state
nnx-pytree-v3
nnx-refactor-custom-vjp
nnx-refactor-graph-utils
nnx-refactor-vmap
nnx-remove-display
nnx-remove-state
nnx-remove-transform-metaclasses
nnx-runtime-defaults
nnx-simplify-flatten-unflatten
nnx-simplify-readme-again
nnx-simplify-unflatten
nnx-simplify-update
nnx-unsafe-graph-cache
nnx-update-transforms
nnx-variable-and-graph-utils
nnx-variable-issue
nnx-variable-refactor
nnx-vmap-cond
nx
pr/IvyZX/4145
pr/IvyZX/4171
pr/IvyZX/4343
pr/IvyZX/4624
pr/chiamp/3782-1
pr/chiamp/3782
pr/erfanzar/4412
pr/gnecula/3861
pr/mcsmart76/3953
quantile-regression-nnx
randomness-guide
refactor-update-context
remove-variable-state
revert-626-nlp_seq_linen
revert-5183-bench_fix
rnn-v2
rnncell-refactor
set_mode
split-pytree
standalone-nnx
tabulate-flops-fix
tabulate-sharding
test_317378007
test_317793206
test_318181863
test_318341372
test_321144725
test_324836484
test_326589776
test_331916531
test_332047658
test_332852862
test_333036622
test_333432938
test_337964891
test_340818348
test_341637550
test_342960801
test_344339107
test_346206557
test_348184387
test_348290078
test_348290081
test_348599074
test_349699757
test_351376151
test_351584173
test_352547458
test_355497424
test_360160709
test_362256497
test_362897819
test_368110589
test_369672468
test_375880589
test_380146238
test_381300586
test_384593511
test_384833512
test_390305432
test_394394031
test_399373571
test_400667551
test_400933467
test_402700753
test_405962065
test_411882335
test_424634712
test_424833419
test_425358614
test_430675887
test_445402695
test_454835318
test_459238640
test_460812552
test_476904885
test_484569827
test_484804825
test_494218037
test_495168957
test_511585331
test_511586735
test_517661790
test_522401382
test_524020065
test_524156426
test_525967287
test_526728576
test_527795240
test_533548381
test_537384891
test_543518398
test_563486995
test_563652936
test_569542141
test_573301464
test_579987248
test_600454908
test_610717214
test_622872682
test_623822696
test_629615743
test_642421861
test_650338576
test_660085584
test_669678371
test_671493323
test_677843398
test_681582933
test_688953662
test_694151533
test_705077026
test_720657969
test_730436599
test_734367472
test_738189137
test_740051057
test_743764112
test_745698612
test_747024381
test_748311465
test_773116112
test_782979395
test_797882059
test_801930691
test_804790020
test_804851220
test_815862166
test_838895956
test_846566054
test_855783481
test_866211786
test_885309371
test_886082903
test_886581793
test_896115256
test_896146741
test_900270194
test_903485800
test_903830576
test_918133843
test_923984648
test-main
tf-howto
tmp
tmp2
to-linen-eval-shape
transforms_tree_guide
updated-pyproject-toml
use-kernel-metadata-in-gspmd-guide
use-register-dataclass
use-seq-lenghts
Merge pull request #4891 from mohsinm-dev:fix-nnx-tabulate-empty-dict-issue-4889
a-googler
committed
250 days ago
f65afdb8
Merge pull request #4948 from samanklesaria:issues/4883
a-googler
committed
250 days ago
ebb980a6
clean up jax.Ref-related names
Cristian Garcia
committed
250 days ago
4c4bc3b1
Merge pull request #4980 from IvyZX:shardref
a-googler
committed
253 days ago
a9dccc06
Clarify `*Norm` layer docstrings: `axis_index_groups` is unused under SPMD jit.
arnoegw
committed
253 days ago
ae1594b0
Remove type annotation
samanklesaria
committed
254 days ago
8b7c8bd5
Merge pull request #4983 from google:propagate-eager-sharding
a-googler
committed
254 days ago
34e1d68e
Merge pull request #4968 from google:get-set-metadata
a-googler
committed
254 days ago
b9cc05f4
Merge pull request #4981 from google:jit-wrapped-types
a-googler
committed
254 days ago
609a0aba
ignore uv.lock
Cristian Garcia
committed
254 days ago
f8c5943c
propagate eager sharding
cgarciae
committed
254 days ago
7959380f
Support shared structure
samanklesaria
committed
254 days ago
126b2139
Merge pull request #4970 from google:fix-typos-flax-gspmd
a-googler
committed
254 days ago
7e8f1522
Merge pull request #4982 from google:version-0.12.0
a-googler
committed
254 days ago
8fe45a76
[nnx] preserve the function's type information in jit
cgarciae
committed
254 days ago
566daf7d
update to version 0.12.0
cgarciae
committed
254 days ago
960e52e6
Use localized YAML dumper to avoid global side effects
mohsinm-dev
committed
254 days ago
Verified
fee117ae
Move arrayref creation to the end of variable creation
IvyZX
committed
255 days ago
6fab0a37
nnx.summary: simplify tail padding and rename test for clarity
mohsinm-dev
committed
255 days ago
Verified
aa5f3467
Copybara import of the project:
IvyZX
committed
255 days ago
f400015e
Merge pull request #4967 from google:variable-disallow-implace-operators
a-googler
committed
255 days ago
8bb57846
Add check for duplicate nodes
samanklesaria
committed
256 days ago
44ed2fd9
Add compute_flops and compute_vjp options to tabulate
samanklesaria
committed
256 days ago
27ed2531
add Variable.set_metadata
cgarciae
committed
256 days ago
0e2b0f33
raise on Variable inplace operators
cgarciae
committed
256 days ago
a4ec6a5d
Apply suggestions from code review
samanklesaria
committed
256 days ago
1bd6b075
Typo fixes in flax gspmd guide
vfdev-5
committed
257 days ago
Verified
e1febdad
Copybara import of the project:
epiqueras
committed
260 days ago
49f51162
Merge pull request #4969 from IvyZX:cfg
a-googler
committed
260 days ago
939f5323
Merge pull request #4966 from google:fix-pathparts-import
a-googler
committed
260 days ago
a6ab4d7d
Newer
Older