Wheelodex » Projects » flax

flax

View on PyPI — Reverse Dependencies (184)

0.8.2 flax-0.8.2-py3-none-any.whl

Wheel Details

Project: flax
Version: 0.8.2
Filename: flax-0.8.2-py3-none-any.whl
Download: [link]
Size: 686754
MD5: 620eae2a401547625eef7e764fb0b13f
SHA256: 911d83e01380fdb3135c309e70981aabd15e7ca038014d7989ddc3cfaf4d0d45
Uploaded: 2024-03-14 11:35:04 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: flax
Version: 0.8.2
Summary: Flax: A neural network library for JAX designed for flexibility
Author-Email: Flax team <flax-dev[at]google.com>
Project-Url: homepage, https://github.com/google/flax
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.9
Requires-Dist: numpy (>=1.22)
Requires-Dist: jax (>=0.4.19)
Requires-Dist: msgpack
Requires-Dist: optax
Requires-Dist: orbax-checkpoint
Requires-Dist: tensorstore
Requires-Dist: rich (>=11.1)
Requires-Dist: typing-extensions (>=4.2)
Requires-Dist: PyYAML (>=5.4.1)
Requires-Dist: numpy (>=1.23.2); python_version >= "3.11"
Requires-Dist: numpy (>=1.26.0); python_version >= "3.12"
Requires-Dist: matplotlib; extra == "all"
Requires-Dist: clu; extra == "testing"
Requires-Dist: einops; extra == "testing"
Requires-Dist: gymnasium[accept-rom-license,atari]; extra == "testing"
Requires-Dist: jaxlib; extra == "testing"
Requires-Dist: jraph (>=0.0.6dev0); extra == "testing"
Requires-Dist: ml-collections; extra == "testing"
Requires-Dist: mypy; extra == "testing"
Requires-Dist: opencv-python; extra == "testing"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: pytest-cov; extra == "testing"
Requires-Dist: pytest-custom-exit-code; extra == "testing"
Requires-Dist: pytest-xdist; extra == "testing"
Requires-Dist: pytype; extra == "testing"
Requires-Dist: sentencepiece; extra == "testing"
Requires-Dist: tensorflow-text (>=2.11.0); extra == "testing"
Requires-Dist: tensorflow-datasets; extra == "testing"
Requires-Dist: tensorflow; extra == "testing"
Requires-Dist: torch; extra == "testing"
Requires-Dist: nbstripout; extra == "testing"
Requires-Dist: black[jupyter] (==23.7.0); extra == "testing"
Requires-Dist: clu (<=0.0.9); python_version < "3.10" and extra == "testing"
Provides-Extra: all
Provides-Extra: testing
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: AUTHORS
[Description omitted; length: 7901 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.43.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
flax/__init__.py sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA 1135
flax/configurations.py sha256=ujrTueyd4ZDZ27M-KZ2zczVHc6sHzZxKN_URMMsZb_Y 5990
flax/cursor.py sha256=mhIa8Czih_kUHfIV5Aq6YbjqutYzyPFi40oolHGmIlo 26339
flax/errors.py sha256=7vnunxbS-Lhc7N5xy0IsNZ-RuDAQGgpF0ENBtFUzVY8 30713
flax/ids.py sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4 1752
flax/io.py sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q 5309
flax/jax_utils.py sha256=FTtXCqYKRoxGTAeLLIYMGn0JFNAph6_LiTl5_cLpL0c 11534
flax/py.typed sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE 58
flax/serialization.py sha256=LBi2n4uCXHpUQ9HJsiIjnlHu_Ae8y1S1QldiwuOtymM 14315
flax/struct.py sha256=H7pFpf0GEFKdJnaITdunPzoFY7apUIMGF4oE5zDl1oI 7886
flax/traceback_util.py sha256=VSBwID-DMPnPfXlESEkNlqZIVj009mF0kkdlvho8qXE 1990
flax/traverse_util.py sha256=UGSxF9PK8Io2NXyJsrFy8ahYLsxXyfHeKVLMWQi02eI 13746
flax/typing.py sha256=d_YGzVgJAjVGLWy1fPFVpO7h3YE_U8bjLQdSX8L81-g 2560
flax/version.py sha256=_7ACLupePNI1h1R6JIV1CGrJvGxw4OLlbUXdawFSHks 650
flax/core/__init__.py sha256=9gthSYfQPBSZvkUNAF6Rl6TzkUJqrEMcL1h4HqPQPWA 1467
flax/core/axes_scan.py sha256=dDT7xT-3fRY75tit1SXFFudMsKWcDnyfrUKk699YkVw 5359
flax/core/flax_functional_engine.ipynb sha256=DGo3J5pSMS6BaDhQKUopSAt6NOQ8dp1lF0v8w63clMo 10482
flax/core/frozen_dict.py sha256=PvlvO9xTJGG_1s8KxCFuwsxTPC31-D3W1Sf_YrPyGco 9974
flax/core/lift.py sha256=pPt1g-TeU7sMQSmV6LCv7bb3TGrnicjkUdrC9Y9QUrg 61453
flax/core/meta.py sha256=74tfTByQRuV-BELuNrSrCia-aq6jnhMFnbTKEHTMgRg 11745
flax/core/partial_eval.py sha256=hj_llMsUSSqH1eF6bUdnhWsWFpLlApOPj6Y9d7VbnOg 2546
flax/core/scope.py sha256=qA9ZG4lDu6yzAWF9ug3oIoRp4CBZ-BisFWy_BWWSX-M 38092
flax/core/tracers.py sha256=wzi47_1VH5SbozYVfnvtUfntbrOe2JDlKEjBM0wOwkg 1054
flax/core/variables.py sha256=KAkQwMGEY-fASlJPHcGpk7vA-8B7VBMlasAus-QfA74 1558
flax/core/nn/__init__.py sha256=2YpgoLw5eP4npw6MOhgUdfMI4nrGIKfC-fsnDNuULJY 1795
flax/core/nn/attention.py sha256=jlaJ0m42bjslMZQRK_0l2yYV8x1bY23pQGoxLqUziwg 17932
flax/core/nn/linear.py sha256=iuOteK4JoyDCQ94O3oeC0Uo8yyxJ84IRurHRU7h-_z4 12040
flax/core/nn/normalization.py sha256=z8LEGKyWitdq73S_j1WlTa7P5uNJSGL-Qm_eulrqDfk 6897
flax/core/nn/stochastic.py sha256=svzeqhl0XBLKP2t7Ewgc739xuItFBZvnk7oGjlZSpfo 1488
flax/experimental/__init__.py sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE 582
flax/experimental/nnx/.gitignore sha256=8cCwBcCnXagyLTaxk0142C5wUEODVjMB6qVOPhp_lSc 1831
flax/experimental/nnx/README.md sha256=W_JspIoKADI3vFCoQHYS-NtGePdw5K0nLuo2ZcijTgM 21176
flax/experimental/nnx/__init__.py sha256=OEUZoxVWBngAoAQmttzlqqSyd-f9pWHWFDgJT_kxT8I 4909
flax/experimental/nnx/docs/blog.md sha256=mZY76YGZtxQLxnm2JgEgGXa1KFqGgzqFDq8ojoimvxw 350
flax/experimental/nnx/docs/demo.ipynb sha256=67QPe0rf6Ok10DwI30FoZ8fok6XiFoukXMggchrUcpk 10272
flax/experimental/nnx/docs/demo.md sha256=PK6xTPWMJ0n4175HCwto2zI3egXhy48DFVneqxtFsC8 4450
flax/experimental/nnx/docs/quick_start.ipynb sha256=VGwguPgGDqE97APZfSzwyctkEBqkZ4ME7xcMaYhL4NY 61270
flax/experimental/nnx/docs/tiny_nnx.ipynb sha256=QlYdDKUqJXm49-SZ4LXDMolf9OCZlhE5_qDybXUvQvw 16288
flax/experimental/nnx/docs/why.ipynb sha256=5SQpOY-4zNQ-4B7DunTABTGfkGXIDP1JUCC4Tuf31eU 26897
flax/experimental/nnx/docs/why.md sha256=-q6Q3_BaOxQY644FervSr8kBi8lvGc9DSjBgKkj74vw 15217
flax/experimental/nnx/docs/images/stateful-transforms.png sha256=pN1xMyF1Pj7-Dkh3k59E72-YzyMVPiXKC5IHDf21yRk 304812
flax/experimental/nnx/examples/lm1b/README.md sha256=1fBFkKf0NHuVYthf9Pcj5HwjLAVbgJk0tEF-36Fg3bg 3267
flax/experimental/nnx/examples/lm1b/input_pipeline.py sha256=-dqS1rhL0LPXHWNdV7_kOdDjF4yUzBYfOpKtlleOMjk 12344
flax/experimental/nnx/examples/lm1b/input_pipeline_test.py sha256=xbsaCa9vNPfXLQV2s5zXh2wzELIZ4CQqzv4-TopAY7s 3293
flax/experimental/nnx/examples/lm1b/main.py sha256=k28k3dKrQQVeQHYoJ8Yi-XyiJDNGz9r10uDTUaCq3Es 2143
flax/experimental/nnx/examples/lm1b/models.py sha256=YSWXawFjQTl2UEOzScBiE_LD8dsls8m_MWQ9Azt1wIc 15055
flax/experimental/nnx/examples/lm1b/models_test.py sha256=V8sO9DX_0PzhKvWZYQ96AaLPVKt-cuCSxgfWsFt3PVo 9517
flax/experimental/nnx/examples/lm1b/requirements.txt sha256=aXtwCM86ZdiKomhmGETw8rcNEo2N9rF2MXWHyuMBqxc 347
flax/experimental/nnx/examples/lm1b/temperature_sampler.py sha256=tyoiOf4dwoLqkau4tOp7hilU2ocmafSDPYhFeo7oFX4 4799
flax/experimental/nnx/examples/lm1b/temperature_sampler_test.py sha256=ATe15vuNrU51PlJVdablJ60Xn4hkOptUtrCgye5PaR0 1448
flax/experimental/nnx/examples/lm1b/tokenizer.py sha256=JoQD9UhDe_ff7UnoAfrbvbL4Z-FDj1ZFRgwr3_x5_r4 5263
flax/experimental/nnx/examples/lm1b/train.py sha256=piFKdcucWR6fHYvDOtEPvhsSk2lqHNukmrKgUtm2aBo 20465
flax/experimental/nnx/examples/lm1b/train_test.py sha256=huVCYOMfppbJWNhTpnJaTnrtQWZ7sjjDHEz-KrWc3Zc 2026
flax/experimental/nnx/examples/lm1b/utils.py sha256=lBPJXXdjjVDx9Cb6rL9uB5sEMirRCE4VHxdvM32kS3Q 5015
flax/experimental/nnx/examples/lm1b/configs/default.py sha256=rp8zsTa4a4WK_29M57cletqNk6eDsT6xYmCuML24zJ0 5096
flax/experimental/nnx/examples/toy_examples/00_demo.ipynb sha256=2uSca1EJg736IqQ-kbBHYzVPFQ2aa5y3by-d1n4zqfM 13748
flax/experimental/nnx/examples/toy_examples/01_functional_api.py sha256=qqbVkdzDqN7Ze_PNmQnAajbz8SpSK-GJW3uA-0_eibc 2814
flax/experimental/nnx/examples/toy_examples/02_lifted_transforms.py sha256=wbH8psSaRzigs8qlmWT1wp5VZIiTa_Kx41aIJ3_apvE 2571
flax/experimental/nnx/examples/toy_examples/03_train_state.py sha256=fahgGzDIJLulJOIqhXwrLKpAIM936qdJbWdkRRolQ90 2872
flax/experimental/nnx/examples/toy_examples/05_vae.py sha256=RW0UFvKAB1k6HHVZSBSmU0OMmBb2JbiOiCWNXmyoUhA 5704
flax/experimental/nnx/examples/toy_examples/06_scan_over_layers.py sha256=ZOqvz9B9adRaDfUQw4dFM0wnvtxVNB0wDbAEZuTYiVA 2785
flax/experimental/nnx/examples/toy_examples/07_transformer.py sha256=fpI8mj5HgCXsFTfd4trQk_D-tUvRpstBGVZfQEJHAcA 12131
flax/experimental/nnx/examples/toy_examples/08_save_load_checkpoints.py sha256=HxczamIuiXm4cWrE-wn4SrOeEe49TZy5xsI8w8cKYk4 1936
flax/experimental/nnx/examples/toy_examples/09_parameter_surgery.py sha256=69coIqTsRuEz4yJ7Tm_t9dlXjMtOa4lBMvBkQg4xwBg 1940
flax/experimental/nnx/examples/toy_examples/10_quantization.py sha256=EyhqfQ-SzbYSAALnSObfFu78T3BQhRYoL9FLeePlowE 11038
flax/experimental/nnx/examples/toy_examples/requirements.txt sha256=1HrvOupPDAFMS_9A7CLrwMhdwU1_d9bRP4i3PU27KjU 35
flax/experimental/nnx/ideas/shape_inference.py sha256=Q6hdqCJLDJBUlVwOgTaPb1_24NXhFT7veOi-YldLsMY 5520
flax/experimental/nnx/nnx/__init__.py sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg 1163
flax/experimental/nnx/nnx/compatibility.py sha256=0c8GMHjyDMQjyB4HYduN_QAlmBhhqk5iCqYH6KxeuCY 3134
flax/experimental/nnx/nnx/errors.py sha256=5FCT85_hNWEiikufgReDuwN_uC_4HxZj-VMHS7mIcOY 626
flax/experimental/nnx/nnx/filterlib.py sha256=h-Hm529II5ZyHEK7kkvghvMg8aEc8lAfPC61z1H1oK8 2631
flax/experimental/nnx/nnx/graph_utils.py sha256=uBD-Kqwi4py4Q9o5PKJlzrJmf_LJVzfDJPSN4pqMpWc 25695
flax/experimental/nnx/nnx/helpers.py sha256=VVoyXj9LZjMMsa0qUNjXjRlztOdlho5E6KDtoGVG07o 5598
flax/experimental/nnx/nnx/ids.py sha256=vuURWDY8w7XWiRpUz3njkiVEciJZY87mrcmSTc07mjY 2326
flax/experimental/nnx/nnx/module.py sha256=leBZtBh_oPgBVt0HStVrzoq-IcaPqVI3kGOnb-vvXxw 17574
flax/experimental/nnx/nnx/proxy_caller.py sha256=cgiCr36VJSsMeSsamLJDk0ry2MTCzFznBgjEZVFmaXc 2562
flax/experimental/nnx/nnx/pytreelib.py sha256=DdwuQ0A4bJ-bQx6nUdWsukAcAChppGJPcIag7A9L48I 8721
flax/experimental/nnx/nnx/reprlib.py sha256=APQG58t4WPgJJsasH5SOvTPt4jChKkZo6ChWIS3PzM8 2781
flax/experimental/nnx/nnx/rnglib.py sha256=61q01X9i3mChHv9u_LrdWL9kM-ywrOOglEVF92RJYmY 7296
flax/experimental/nnx/nnx/spmd.py sha256=9ol8s_WJBjh4mC_YptZGEP-qBR08b70zlG650Zmmqvo 6410
flax/experimental/nnx/nnx/state.py sha256=zzzLVB1yPMWpg3MgKEGDyH3TS2XDhfmnxB8rHrxK8z4 7556
flax/experimental/nnx/nnx/tracers.py sha256=krybaDx7yG0RotXgFBJkPjplLRzAm-umS5WxXyvtf-U 1728
flax/experimental/nnx/nnx/transforms.py sha256=zsdqXQONmZZp_l3e9In7DCyvlia0RwHGQBHtTzxF5fw 44735
flax/experimental/nnx/nnx/variables.py sha256=T2GguQleJn7b6CcFgl3dlB0hUv_ofRFLGfMpSuvHdBA 15128
flax/experimental/nnx/nnx/nn/__init__.py sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg 1163
flax/experimental/nnx/nnx/nn/activations.py sha256=Ub94GPKCDh4S4ejqP3r8y3Znh9To78APHCs5AUB-K8g 1219
flax/experimental/nnx/nnx/nn/attention.py sha256=bIixGAKkF1V0URyxEH8D0lAjn51HR-SfZbFgkysZXoU 25288
flax/experimental/nnx/nnx/nn/dtypes.py sha256=zEeCbEXsCeAFvGbwCR6gZx_KM70bQ_15S_uljb55aaI 3318
flax/experimental/nnx/nnx/nn/initializers.py sha256=uOyOtYl4IbsPemyoRONbzHjj00pvvpSmmZecdaW_trA 2527
flax/experimental/nnx/nnx/nn/linear.py sha256=GKwFHNiWlJjndD6TXOUDOtuZ5XF7MB8DpGFcCZgr73Q 22682
flax/experimental/nnx/nnx/nn/normalization.py sha256=m_n3Py0cTG45JkHCu0texxks6lUAeyTQ7NW1sWxmQTE 18368
flax/experimental/nnx/nnx/nn/stochastic.py sha256=8svtGy9G3NgZIEF5rZ8Oj43lsgZMoM-pxI_EjcGrth4 2800
flax/experimental/nnx/scripts/requirements.txt sha256=xxQ4qS5Xhzyxa4FDTLP4vvIvM8EStbTyLWpAMwKLH9o 17
flax/experimental/nnx/scripts/run-all-examples.bash sha256=DyFMZPSCVL7WG4kcUxN0P9pIcuoJWAMil7UkB3BiMzE 285
flax/experimental/nnx/tests/__init__.py sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg 1163
flax/experimental/nnx/tests/test_compatibility.py sha256=7JGj2B3cc_TIOcNjaGTVQiIyfFYk2EaElZKr1cJIcOE 1204
flax/experimental/nnx/tests/test_containers.py sha256=P9kpOSJ9kxM_TI5vkZhBfh9Mi1c2z2MCGoG5T5kZ3OM 1636
flax/experimental/nnx/tests/test_graph_utils.py sha256=VOXlLWtU3YAWgGvql6dPOXxjgWHfEzDw9aBm7x13Z9s 7433
flax/experimental/nnx/tests/test_helpers.py sha256=a2FfI54avzM8TCLK4tmH79S_8QKaHl7edWkt5xYFEvI 2078
flax/experimental/nnx/tests/test_ids.py sha256=iyzxi_ene5RGX7r9EBZl2XJTGQ3CEPZNsiLKqFeebfE 929
flax/experimental/nnx/tests/test_integration.py sha256=8JsIaUpcaxKt1vg2PHUW4-AXZ81l8mbp-3SHFcsOb1g 7368
flax/experimental/nnx/tests/test_module.py sha256=dO7phGdakfHdP0kgMoHf92gigZvtfc17djjORL46g_Y 15907
flax/experimental/nnx/tests/test_partitioning.py sha256=D0SnN3vu1W8wrNkoWtzQO3f66CSGDeTo67sxmWsn1yI 4232
flax/experimental/nnx/tests/test_pytree.py sha256=604BnPbNo7bsXzYxiJOjQrs0VaywgOhH-BQWEdaBpRc 6641
flax/experimental/nnx/tests/test_rngs.py sha256=Y03uZe4A1HbKobSxPJjALzpgEIHgBy09D46apUYXlr8 4699
flax/experimental/nnx/tests/test_spmd.py sha256=A9acC-ETBEZI8wllBTyMGBlrFncK6RLNsj7eC-tFyyY 2184
flax/experimental/nnx/tests/test_state.py sha256=UnvTTCzP8stJgqaK-r3wyfRpA_7xTGsCwChaNEuzkz4 2077
flax/experimental/nnx/tests/test_transforms.py sha256=XasBexZIrIWJoEZcjLn7eqx95xASrgnbC8RMcZ7hL_4 18175
flax/experimental/nnx/tests/test_variable.py sha256=l02oH-ieRvDN4sPHpDC-Vo-Qq22mwPOwXCi9oWjdwSU 897
flax/experimental/nnx/tests/nn/test_attention.py sha256=4sPGXHQpMq7H93Z1gyH1e-0c50zssln_Bw83rroa9wQ 4937
flax/experimental/nnx/tests/nn/test_conv.py sha256=BR_FUuGCNMGfBUNkjK5-LfMCNbd7dcOL9OnX_mPJk24 3003
flax/experimental/nnx/tests/nn/test_embed.py sha256=NXJ47iH6WYA9UjGsspw9DcPpZROihrRUT_mX0dkACAo 2077
flax/experimental/nnx/tests/nn/test_linear.py sha256=oyqqtglUkF1liVo_Gokew1fJmU76CsxIWhNQ3Du_jQQ 2643
flax/experimental/nnx/tests/nn/test_normalization.py sha256=BnUDtMmvtNc_2-fUMaT-uXRzBYuxZXkRAPJLTl5G3yo 7537
flax/linen/README.md sha256=fQOTJDyHjMuIUiBxOomju528Xfolp3oGBo9q5l9U-Uk 2191
flax/linen/__init__.py sha256=BjlQtBXhAEz-lJsP_U5I0HTbuZvLA4FXeTtOE7duItA 5253
flax/linen/activation.py sha256=LeFjjROwX7GLlyfdA-iBaRNY_qp1PKhGnir4ggxAVV0 4101
flax/linen/attention.py sha256=R3B2jiWNlXI_l9EQfkdq4VKaLvvhH053LPvjN1nfwA0 31104
flax/linen/batch_apply.py sha256=7B_bwfI2vJyt2UPMtOcjEo6puLFavcJeLrF-COoAntc 4172
flax/linen/combinators.py sha256=WW-GSBw19Wwv0Mpg_VOdYqSmH7-8loAqDkj5DJ1xcNA 3846
flax/linen/dtypes.py sha256=hyXSAKgx6iK15cXrojy4f_DHzSq160Jdv_1Mg7vqoaM 3860
flax/linen/fp8_ops.py sha256=KGyy1tgHXUqbeN6a4qe_hUJzRUGA-gjLEyO6VJb9Sfg 9324
flax/linen/initializers.py sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o 2676
flax/linen/kw_only_dataclasses.py sha256=U4-mGzgAhz902A1SL29J8Hdr33smxl8NUB9BwOz-4GU 8828
flax/linen/linear.py sha256=K_cc_zQF8HZc8zXkY1uCt9dP2QHETiZvMFvt0rIiFC0 44278
flax/linen/module.py sha256=HyOsrQnQvGeeUmRw1-onOi8C_mzFabGYgnj6z2I3Yu4 109421
flax/linen/normalization.py sha256=Y1INtrFDmT2KLNbI0831DlXjIEbqE4xJO_Ux_BKIoaI 48880
flax/linen/partitioning.py sha256=60TRq8NzoeN8992cjTbGIqQxwmEyEAqXHrgol8ncACo 19421
flax/linen/pooling.py sha256=x6BACBzHgMrCB3xAIRbp0bYmPL7ux_sIce83qvkggiQ 5554
flax/linen/recurrent.py sha256=9voiS4zDBjijV6rt3wyvJ6YahZbIHoSWhcqexg7Grh8 45635
flax/linen/spmd.py sha256=IubWjM85LqHjWnIb9BoNFIap8vTVNocV8cEcOzLqMyc 11192
flax/linen/stochastic.py sha256=tUthbCWYuSfMW_dd397l_IRxDQyujsQ_4FRNcxSiGGY 3619
flax/linen/summary.py sha256=26LfD2Rm1prX2Eehgly1dA6w9wHL6UWAfjtZ2BwEK7Y 26065
flax/linen/transforms.py sha256=mG1atFIwWCeYMRIGuB6iF4M7ZVdv0FN-liMY2nvvfpM 79055
flax/linen/experimental/layers_with_named_axes.py sha256=xv-vBWVgIB2lfHF4KrryM66jQkgjYgeC9qo2Ansg7K4 11235
flax/metrics/__init__.py sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg 1163
flax/metrics/tensorboard.py sha256=HUDQYfdhZXTCCxQJMpr9aUzrqexGZasEGvzu9ccy-ug 7792
flax/oss/ .git-blame-ignore-revs sha256=eJmLQj4P4rVEN8uRkflxwOhLFte29Y0WeGoI1uRNWfY 443
flax/testing/__init__.py sha256=UarXpjywrTaGYJBYPf0joEzJawWavRs_f58YZJowfh0 647
flax/testing/benchmark.py sha256=dumZs_E0VmGqiEQHwe8eOzxgv5bjWtXj5NKqrnsmF7E 9320
flax/training/__init__.py sha256=uJTipyYX83XazXl5bkf17EF4DjJE5s1YhCA3vS0vXqo 613
flax/training/checkpoints.py sha256=P56XV8DBqjm0be-UNLRlsIHotfw1iHIG2N9l4lNcLIc 43940
flax/training/common_utils.py sha256=_3-gAQuFBC6ITUiu9dgGMZu0PS6SLlzd59UlnwfHAJg 3690
flax/training/dynamic_scale.py sha256=YD1Aby3DqDlfi_KKkdVuW4gBi_rz0K2-3xIj1D5VHXs 6056
flax/training/early_stopping.py sha256=uy20C4GxzYI7GV0re-vYH-Itj9f_u0pf6M2uZPe8IgA 3194
flax/training/lr_schedule.py sha256=SYCWUEgl_iZyOZpIFaP2wzIgniamzu6I1MDS1R4Hlsg 7419
flax/training/orbax_utils.py sha256=HL_6WozkC-5cEUQFy-yxnH22PlV5m_v8pfbSOxO51Xs 3685
flax/training/prefetch_iterator.py sha256=KnhVdWbGC--2YTaxKeuGJTfH2TrvtjQWP3mk3s6KHZQ 2978
flax/training/train_state.py sha256=Hpm2kag2BtLh7PXoYP1xDCwo1vyb28bJXwiCt6vb3Vw 4731
flax-0.8.2.dist-info/AUTHORS sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE 293
flax-0.8.2.dist-info/LICENSE sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY 11309
flax-0.8.2.dist-info/METADATA sha256=rmK0OeiA3YEa6sbUu6dmGPy0z-NuSGm7oxCqjN53nxs 10196
flax-0.8.2.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
flax-0.8.2.dist-info/top_level.txt sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM 5
flax-0.8.2.dist-info/RECORD — —

top_level.txt

flax