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 |
— |
— |