Wheelodex » Projects » flax

flax

View on PyPI — Reverse Dependencies (185)

0.8.3 flax-0.8.3-py3-none-any.whl

Wheel Details

Project: flax
Version: 0.8.3
Filename: flax-0.8.3-py3-none-any.whl
Download: [link]
Size: 685879
MD5: 0db73e01b548f504413a2d5deab3fcd5
SHA256: 87933bc2aa5e70e92ac227a9bd2adeea6b9960a84eade18139a534851ddaf91d
Uploaded: 2024-04-30 09:57:17 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: flax
Version: 0.8.3
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"
Requires-Dist: penzai; 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=7HtyMIHYXSW6LR3F8vo1RXuV8DNPHqCdNdJ12fLVvlI 26399
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=W1dUVyzmlzX6OJxq618crbCe6eXcs2caFQlDnr0i4Bw 8275
flax/traceback_util.py sha256=VSBwID-DMPnPfXlESEkNlqZIVj009mF0kkdlvho8qXE 1990
flax/traverse_util.py sha256=UGSxF9PK8Io2NXyJsrFy8ahYLsxXyfHeKVLMWQi02eI 13746
flax/typing.py sha256=1VBg_Y81vLTn6rgkLzoUx195Fmxq8K_7W8lv1HZzutY 2819
flax/version.py sha256=YPI4PdcC-mhXT9_pHPPMx-EoXYsk5P_ENRsTVYV91bk 650
flax/core/__init__.py sha256=9gthSYfQPBSZvkUNAF6Rl6TzkUJqrEMcL1h4HqPQPWA 1467
flax/core/axes_scan.py sha256=g0tIO9dF5FpDdkDyByOt8n5lqlyWjHJYWUee-JB-jUg 5763
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=Qjt3NPX-GBBxktmvbLzHUAgbu_51sv9SzI8sempy44E 61688
flax/core/meta.py sha256=PhRKK1ikGeVRdnfK0r-A8KxdyLe1VAbSOCCOKzdz-5k 11725
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=GHHRWXkgv-meQ5cETw0nKrsgsrCbskpUxAJCQzbfcCI 3678
flax/experimental/nnx/__init__.py sha256=IAhMriYSLkrhSiFENLcr0E2M7CJxVHpQXjRVBTxrLsE 5733
flax/experimental/nnx/docs/blog.md sha256=mZY76YGZtxQLxnm2JgEgGXa1KFqGgzqFDq8ojoimvxw 350
flax/experimental/nnx/docs/demo.ipynb sha256=_676MwywHpjp2EyrcpQS4i3seQjyRj7M001KPeOxvD4 10253
flax/experimental/nnx/docs/demo.md sha256=UN1PCtsrgfvOnXsdYMgj0-WjJrkFJvClhU3SUQIKEcI 4271
flax/experimental/nnx/docs/quick_start.ipynb sha256=KvzpZQak1a2cbhAxXEGU_V5G7YzGr47xEYWKgyN2AHk 61295
flax/experimental/nnx/docs/tiny_nnx.ipynb sha256=mVoPLE2onwOpXggPIYh-9pqhxC6IAbcVgjnKExJAufU 16318
flax/experimental/nnx/docs/why.ipynb sha256=ZxtmUBQ0UZ3EIanzGXqMO3-I-SIdsZs_yw3MIzVaAvI 26920
flax/experimental/nnx/docs/why.md sha256=D7_vp-t-Bn73n-pBCdL9BdRxFkQReo2BaqOmsqmxYEw 15238
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=QHQttfoWULAcfcyrTyILOjAbwOD2gbWV1usfikXJWhw 9662
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=UiMD9Zpc6lxSMG9hx3a1kiwcxKsQL8br6eZfjECm2iY 20551
flax/experimental/nnx/examples/lm1b/train_test.py sha256=huVCYOMfppbJWNhTpnJaTnrtQWZ7sjjDHEz-KrWc3Zc 2026
flax/experimental/nnx/examples/lm1b/utils.py sha256=vzm81KNUYnsj8wYNsB4pWHw9lnbUcmCwM-geZaK2Xbc 5036
flax/experimental/nnx/examples/lm1b/configs/default.py sha256=rp8zsTa4a4WK_29M57cletqNk6eDsT6xYmCuML24zJ0 5096
flax/experimental/nnx/examples/toy_examples/01_functional_api.py sha256=-4p5LOkCBJszcvf9BRVd1hHgsGFb-08tH0uumBjL3Qg 2868
flax/experimental/nnx/examples/toy_examples/02_lifted_transforms.py sha256=NdejwojFQ3GgQMEnxWdcf9bZOC5HXne02OnJ1V_yHfc 2593
flax/experimental/nnx/examples/toy_examples/05_vae.py sha256=2jn0hiFHnPxk70lz2s3dCxj20xlFCnuZJ4i1zF5PRA4 5208
flax/experimental/nnx/examples/toy_examples/06_scan_over_layers.py sha256=uJR71_HdVPntZtW3iuYTp5uFOdh_P6BEVtXK1B3vAMM 1865
flax/experimental/nnx/examples/toy_examples/08_save_load_checkpoints.py sha256=FdwehZ02FKEDFjbeS3EJCV2S0kv9CW10w68q-oMsL-U 1946
flax/experimental/nnx/examples/toy_examples/09_parameter_surgery.py sha256=BQh6L6FG1eEqeKNYL0vEiFO4egcpkEF-Zv_S5bZROTI 1969
flax/experimental/nnx/examples/toy_examples/requirements.txt sha256=1HrvOupPDAFMS_9A7CLrwMhdwU1_d9bRP4i3PU27KjU 35
flax/experimental/nnx/nnx/__init__.py sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg 1163
flax/experimental/nnx/nnx/compatibility.py sha256=OI_KMyOVGVsKUam51ByDOYmX-18RY8ZN2mfkZh_6FbM 3187
flax/experimental/nnx/nnx/errors.py sha256=5FCT85_hNWEiikufgReDuwN_uC_4HxZj-VMHS7mIcOY 626
flax/experimental/nnx/nnx/filterlib.py sha256=CWxfA--U6khPxUlYUDmL12s9jIDu7oQsd446fAW3t-g 2913
flax/experimental/nnx/nnx/graph.py sha256=NTXA9e0I-fO7E9VA_LqHiVMLMcHF48_CVsmLZgZba-w 37769
flax/experimental/nnx/nnx/helpers.py sha256=9D5DRCxCTquxPf-402gwiUtDaWQCHdfyoeMxqV6upa4 6528
flax/experimental/nnx/nnx/ids.py sha256=vuURWDY8w7XWiRpUz3njkiVEciJZY87mrcmSTc07mjY 2326
flax/experimental/nnx/nnx/module.py sha256=7hTyws7cBBgnu5cBOQbwztoSlemIXWLFMFa640eQdk0 12260
flax/experimental/nnx/nnx/proxy_caller.py sha256=cgiCr36VJSsMeSsamLJDk0ry2MTCzFznBgjEZVFmaXc 2562
flax/experimental/nnx/nnx/reprlib.py sha256=5cf2wYupTwV-eRuxKZzBDUABZBSBDHc9s0Jc2uU5Zo0 3200
flax/experimental/nnx/nnx/rnglib.py sha256=7uxF3FnKbD_MKPtFnWDAjr9VYSDBAYrYV_AxTl9YmVc 8995
flax/experimental/nnx/nnx/spmd.py sha256=RS-siiA3ZcM2j7H9ZUuEriNneoEIhrXIBikTsObVY_U 6373
flax/experimental/nnx/nnx/state.py sha256=6UKTMzaVQNlIheEr8sq1kKbw2hd2ZuWOZYtunjdbZwE 7560
flax/experimental/nnx/nnx/tracers.py sha256=krybaDx7yG0RotXgFBJkPjplLRzAm-umS5WxXyvtf-U 1728
flax/experimental/nnx/nnx/transforms.py sha256=eTPkQqldWjGDvc9DK-U9BNAJhtWD5RjFE4oNJunolso 55991
flax/experimental/nnx/nnx/variables.py sha256=R7DLQipWtA-_lfh4yvxYpXqtxlhhuoNkUOz6G-hV0HI 22081
flax/experimental/nnx/nnx/visualization.py sha256=R_u-CXhT28rKbQVSSDhnwm0Bolo63b1wOoXeKFyXP1Q 3508
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=KpJseZuse58x-HN89KvRmhjzUfAlGxRZsp4Reh37CCU 25351
flax/experimental/nnx/nnx/nn/dtypes.py sha256=zEeCbEXsCeAFvGbwCR6gZx_KM70bQ_15S_uljb55aaI 3318
flax/experimental/nnx/nnx/nn/initializers.py sha256=hYspmsHObOsGsw1qUG05YzVPp7lycJ1LkFmOIah85AM 2596
flax/experimental/nnx/nnx/nn/linear.py sha256=00oXCj7yYWehIqRaiaqDkl6u5a6vjxeIFSQK6pwfcgk 28389
flax/experimental/nnx/nnx/nn/normalization.py sha256=KDOqB2hO6aUUH5g31PImjzt_pTrQ5QNqPm1arTkTAoo 18356
flax/experimental/nnx/nnx/nn/stochastic.py sha256=R40cfuxvTIlp-2jCNjp9F1OrxgIiSMoWmAWCR8Bqe2I 3520
flax/experimental/nnx/nnx/training/metrics.py sha256=2FQjMMVPTqwN50CeuJ7uUF66Se8zGq_xio-XmDGeI5Q 5291
flax/experimental/nnx/nnx/training/optimizer.py sha256=rg0Nws43IcTKSRwKud24z1Zo4VboUDFnzl1ouy-Yi58 4861
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=FvovXskfQ1PwIbQ9-_-UL5-ccdIajqSxH1NmoJ7YhrU 11196
flax/experimental/nnx/tests/test_helpers.py sha256=_Kpcr1X7_U93Yso9G0qAVHdSWrNzqJnFHrN4AAoDzdw 2003
flax/experimental/nnx/tests/test_ids.py sha256=iyzxi_ene5RGX7r9EBZl2XJTGQ3CEPZNsiLKqFeebfE 929
flax/experimental/nnx/tests/test_integration.py sha256=Lbz9IfnD5KxIuH9i-FEykxH1p-AsD4sCBS3aHBiJCpA 7462
flax/experimental/nnx/tests/test_metrics.py sha256=DprivD_osg85LzNVua_CM5UlcyvzmTLD_9djCQ22Mjs 2351
flax/experimental/nnx/tests/test_module.py sha256=rqx0gMFK-GDqIIBD4RJKtlXWTIeiKPQ0Xr8zq1f2Af4 16827
flax/experimental/nnx/tests/test_optimizer.py sha256=kJEDNtEdSy3eQB8bQWbrZLnEKYCKnn3DSRDEUvg5edE 4064
flax/experimental/nnx/tests/test_partitioning.py sha256=D4LgR7SrVXIxT3aDaAT1dcpdDJCPXtHvBzITq7pJ4Dk 4268
flax/experimental/nnx/tests/test_rngs.py sha256=dc-VSE5dO9dNWXUgKC741wHAJocgHUhd9dmGf67ANIs 6492
flax/experimental/nnx/tests/test_spmd.py sha256=H2fzdSEzGwusgVuLc7oKqMckDPODVir87xJcnBtqnwo 2780
flax/experimental/nnx/tests/test_state.py sha256=DoireUJ_lkpokIzAn5weDUXfVGfhBXD3419KZ4w_Gc0 2078
flax/experimental/nnx/tests/test_transforms.py sha256=EgMQvWStTyHzhv9Igq4lzD_YcjqeWlPh2pi4C7PMoH8 28817
flax/experimental/nnx/tests/test_variable.py sha256=R34AEhW0kZ0N33-v_a914iPyJp1Aa1m8Y_K1k98hCNk 1836
flax/experimental/nnx/tests/nn/test_attention.py sha256=sFkF9NFCUdjMT-qsk8Dn-O4Pa311Mw77jlMA6gw2WHc 4930
flax/experimental/nnx/tests/nn/test_conv.py sha256=tG2bbNkfo1iNXbvxz31hm4saXxdhSozg7fCRiPQsxhs 3069
flax/experimental/nnx/tests/nn/test_embed.py sha256=Wj8zium9YyWFtr3jWfpXO-YNMlQ9Hna9xjLoHme8VXY 2127
flax/experimental/nnx/tests/nn/test_linear.py sha256=tDa2Z3JBPdLA9v-ZBgc1UPsx4kJbdyHov8nUZvCdinE 4290
flax/experimental/nnx/tests/nn/test_normalization.py sha256=BnUDtMmvtNc_2-fUMaT-uXRzBYuxZXkRAPJLTl5G3yo 7537
flax/experimental/nnx/tests/nn/test_stochastic.py sha256=4IkMk0tPG-07Q6uqU85OdJm3TCmoEJnOrgaLkJ-wA3c 2609
flax/linen/README.md sha256=fQOTJDyHjMuIUiBxOomju528Xfolp3oGBo9q5l9U-Uk 2191
flax/linen/__init__.py sha256=KaXdquHoURWfE6hyWBXv1C9KwcCpajQjQ9z7Wq2nZAI 5259
flax/linen/activation.py sha256=LeFjjROwX7GLlyfdA-iBaRNY_qp1PKhGnir4ggxAVV0 4101
flax/linen/attention.py sha256=eRy26C110sbQPyxWlhLeagslpp8kDhj_dshSjG5iFWU 32287
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=2VxW17wysBA3dUSm-mJx4NERCCydYDnaRshN97Vplvs 10062
flax/linen/initializers.py sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o 2676
flax/linen/kw_only_dataclasses.py sha256=U4-mGzgAhz902A1SL29J8Hdr33smxl8NUB9BwOz-4GU 8828
flax/linen/linear.py sha256=CNLEMm3CR1930EvEcANWvbaacQPZG9l6wHvsm24Qju4 44217
flax/linen/module.py sha256=gEUsAJJjRN_L2hzR7AOKRYjgzEZ08REIQvvl0BYxxm4 113027
flax/linen/normalization.py sha256=3RCElnm6A_DwcvOuTt7LQW2BtTcAn5bPrY8gxXPdeBQ 49760
flax/linen/partitioning.py sha256=60TRq8NzoeN8992cjTbGIqQxwmEyEAqXHrgol8ncACo 19421
flax/linen/pooling.py sha256=HNrjOv_ul5QEz27nXg69iSIzfRa42c2cbZh-q8Dciog 5567
flax/linen/recurrent.py sha256=bAVP5PuEW7iUfXg8F7Up9Eyg468F7iNZioES8SPeH0A 45995
flax/linen/spmd.py sha256=XxCWLwemCNoV2NYOvGCb5BS1tJdQCI_zvlWmgudMUxE 11187
flax/linen/stochastic.py sha256=GKhyMB4yG5j_Yq-yfgz-0LwpusjY7UDIVa1trOqoKnI 3634
flax/linen/summary.py sha256=Z-0lonFNquOhdwDHNPo2z1HJ6l3H7BRDc_8A4pqcoiA 26100
flax/linen/transforms.py sha256=KwmpRK4uHshn0rxQ459pspDT2PZNfDHtFgE_k0bN8BM 79289
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=ClSNZDqAzeoJWT4T5gi0Mop0HCNW4yM9EL68acs0SwI 43835
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=F5IwUzh15IlZMa56EAa5uK_lehoBj2Ima5haQs9S-6w 4767
flax-0.8.3.dist-info/AUTHORS sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE 293
flax-0.8.3.dist-info/LICENSE sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY 11309
flax-0.8.3.dist-info/METADATA sha256=H1b4cI9kXSMV0OOYYM4ObtBKZD_Yazgd2au8n9t3HbI 10270
flax-0.8.3.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
flax-0.8.3.dist-info/top_level.txt sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM 5
flax-0.8.3.dist-info/RECORD — —

top_level.txt

flax