Wheelodex » Projects » optax » optax-0.2.3-py3-none-any.whl

optax

View on PyPI — Reverse Dependencies (280)

0.2.3 optax-0.2.3-py3-none-any.whl

Wheel Details

Project: optax
Version: 0.2.3
Filename: optax-0.2.3-py3-none-any.whl
Download: [link]
Size: 289647
MD5: d66a5772aa1cae304539fdf08be8d3b1
SHA256: 083e603dcd731d7e74d99f71c12f77937dd53f79001b4c09c290e4f47dd2e94f
Uploaded: 2024-07-09 14:01:50 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.1
Name: optax
Version: 0.2.3
Summary: A gradient processing and optimisation library in JAX.
Author-Email: Google DeepMind <optax-dev[at]google.com>
Project-Url: documentation, https://optax.readthedocs.io/
Project-Url: homepage, https://github.com/google-deepmind/optax
Project-Url: repository, https://github.com/google-deepmind/optax
Keywords: python,machine learning,reinforcement-learning
Classifier: Environment :: Console
Classifier: Programming Language :: Python
Classifier: Intended Audience :: Developers
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Intended Audience :: Science/Research
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.9
Requires-Dist: absl-py (>=0.7.1)
Requires-Dist: chex (>=0.1.86)
Requires-Dist: jax (>=0.4.27)
Requires-Dist: jaxlib (>=0.4.27)
Requires-Dist: numpy (>=1.18.0)
Requires-Dist: etils[epy]
Requires-Dist: sphinx (>=6.0.0); extra == "docs"
Requires-Dist: sphinx-book-theme (>=1.0.1); extra == "docs"
Requires-Dist: sphinxcontrib-katex; extra == "docs"
Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
Requires-Dist: ipython (>=8.8.0); extra == "docs"
Requires-Dist: myst-nb (>=1.0.0); extra == "docs"
Requires-Dist: matplotlib (>=3.5.0); extra == "docs"
Requires-Dist: sphinx-gallery (>=0.14.0); extra == "docs"
Requires-Dist: sphinx-collections (>=0.0.1); extra == "docs"
Requires-Dist: tensorflow (>=2.4.0); extra == "docs"
Requires-Dist: tensorflow-datasets (>=4.2.0); extra == "docs"
Requires-Dist: flax; extra == "docs"
Requires-Dist: sphinx_contributors; extra == "docs"
Requires-Dist: absl-py (>=1.0.0); extra == "dp-accounting"
Requires-Dist: attrs (>=21.4.0); extra == "dp-accounting"
Requires-Dist: mpmath (>=1.2.1); extra == "dp-accounting"
Requires-Dist: numpy (>=1.21.4); extra == "dp-accounting"
Requires-Dist: scipy (>=1.7.1); extra == "dp-accounting"
Requires-Dist: tensorflow-datasets (>=4.2.0); extra == "examples"
Requires-Dist: tensorflow (>=2.4.0); extra == "examples"
Requires-Dist: dp_accounting (>=0.4); extra == "examples"
Requires-Dist: ipywidgets; extra == "examples"
Requires-Dist: flax; extra == "examples"
Requires-Dist: dm-tree (>=0.1.7); extra == "test"
Requires-Dist: flax (>=0.5.3); extra == "test"
Requires-Dist: scipy (>=1.7.1); extra == "test"
Requires-Dist: scikit-learn; extra == "test"
Provides-Extra: docs
Provides-Extra: dp-accounting
Provides-Extra: examples
Provides-Extra: test
Description-Content-Type: text/markdown
[Description omitted; length: 5613 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.9.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
optax/__init__.py sha256=gSf4ACB123bvkE8x-W4GFogR9HL6m4C4ittEZwf0wIE 16723
optax/optax_test.py sha256=8n0a-ijvCEM23-bO7Wa_cgbhKye1keSPS_b1egFekTw 1070
optax/_src/alias.py sha256=gHlgJGp4U8dJbbC2Kkk3EgxBlb7yIkzfmZNJFgrAzAo 80905
optax/_src/alias_test.py sha256=5pphG4XVLj-knELcdmz_i7lGal2gtp6CgsVfwsTBtjA 25445
optax/_src/base.py sha256=2Zw2bMGfS4TYdHc_ThI9S-txwumwWoAJiXqvi1FoWAs 12053
optax/_src/base_test.py sha256=z-cv8FVJrtUVEKvFdA4oInCGxXdzUK4rXFsI_L3zO84 6609
optax/_src/clipping.py sha256=2b98Edu1waORroKAjZmbTIlR7cAxJMZJZBE6IbKmJrE 1305
optax/_src/combine.py sha256=pi3Mp19T2W7zS0K04Kgwfbkb82VRemLPyRmuJNWq0FI 936
optax/_src/combine_test.py sha256=APahLo52tFywrpw1SX4BBnNNDODlfN0Sg3pQKeY2mTw 9351
optax/_src/constrain.py sha256=tYrEjizwVIqtvkXZS9kLf_ZAtmaeqGhXH26dxfWzEDw 1016
optax/_src/deprecations.py sha256=5umkxmG6vo0zgdltDZsHCngV5xE6aq4l7ppl3MSZfHE 1926
optax/_src/factorized.py sha256=zwyWhRUkmezdzCkmWLOZOK0xowGnr5UAJ7vI0HyGUL4 7400
optax/_src/factorized_test.py sha256=HQE-MYuL7-5jCx5DechUCYSyyXMFnD6YT-jpVYX3q4c 1634
optax/_src/float64_test.py sha256=AAFzfZdtG4JIs2UzUVmCnJ9bPl4V-f-CqMbj1ewNHzo 3665
optax/_src/linear_algebra.py sha256=P25EwMoaOYgEn1ECwD8Z2UwAUIAkSEY-fDWxhfz4Pvs 9614
optax/_src/linear_algebra_test.py sha256=UkgPMz2Xxn_OTRRfQPlL6SamRlXHMwLuN2IEGrw2ASg 7220
optax/_src/linesearch.py sha256=ndkuht-R3g3wBRk2-p_mAAyENbZakdqiBUBO4lkob0w 57463
optax/_src/linesearch_test.py sha256=psJEwCJwIIemtCf_NJMUEmWlX-8vHLb-vwW46bUPSC0 23094
optax/_src/lookahead.py sha256=N9cnrwVtTABxxDLyE4kjNMi4xMIjNniWNnBkq1601gQ 7453
optax/_src/lookahead_test.py sha256=JRrTX_mzkD8hsJy7K8tQSb05ziCyj78HOtjlEEQXI9Q 5713
optax/_src/numerics.py sha256=atUjLyfVFNfmMC-Xf0gGxxsGDgEgssrwA-yvPMp4A0I 4636
optax/_src/numerics_test.py sha256=YRMQQU63pjkIuRmuMF5Vf3Wk9AGqcoM7ObExGVXsPuM 4198
optax/_src/schedule.py sha256=acIVF-Pgq50dIDfSto5dWL3RHGuS0b6nfO4WUWJU1dQ 1967
optax/_src/transform.py sha256=8VjNVeAOVyv5YaDX_2jCmstav_FnWoRhESA54zSmzBM 52716
optax/_src/transform_test.py sha256=mAhpNmTVoE30x4hdn_OSsjybtFVQQx1JiaXez7j7pXw 6783
optax/_src/update.py sha256=kYH4w9lBTY1kS3G6IpLpteIF-gLcvimk6kdH3ZxJZz4 3692
optax/_src/update_test.py sha256=23TBGtiYf8SCy0EzGobExY14AmbVXU3mF4VJlp7Xb5c 2874
optax/_src/utils.py sha256=zU8ZSiRuyif4wZ30YSwzEhZyrCM0FMNx0257w7CygtA 10653
optax/_src/utils_test.py sha256=WP_uP-b985zcyNQBKQSMQG6eXcH1zujL1RqzFZIlfFc 9941
optax/_src/wrappers.py sha256=1F1YJF4_oUuA1VUrbgQvyk89zySBfn5ACrCj93-Mvio 2196
optax/contrib/__init__.py sha256=EtNLWwsJkgTuBlnc2Hha6CR6FNqWY4VaWGJ9ftLIHpU 2522
optax/contrib/_acprop.py sha256=Z54VN6Jj6pt3jiPZ6QMoi8LuvXTlKYBH8WgnbLT4JO4 6934
optax/contrib/_cocob.py sha256=P5W3a8KNj0UttUnI2IDVf_DUGpw6Svmu5LMxslCpz3E 4650
optax/contrib/_common_test.py sha256=GzOcxhjvrMPBhBtChGFDCT0ZLSBrmGUwJQt9O-dDCUo 5073
optax/contrib/_complex_valued.py sha256=sadTeMo01UrTN_SICfAEjDwwm0eoXT1-x28MhdKQJBo 4101
optax/contrib/_complex_valued_test.py sha256=Slvp4GBeetux67KeEcwiWubRCxXfwL9GACGAWAx0VPU 2853
optax/contrib/_dadapt_adamw.py sha256=PN1s_PDPjFoo2843mb1J7OkjmDgqvEz5Sd5R03uH46w 5148
optax/contrib/_dog.py sha256=zPko00JS4unBotjusf-WaZYvvwkDuJzi9bZYMUctxa4 12552
optax/contrib/_mechanic.py sha256=RiwMEdzE3r_75dkzE86BW6zGzWIQS2gM1O5DNOZoZ28 7414
optax/contrib/_mechanic_test.py sha256=XZ27Jc3XpV5qv9WivQI0o7LFyCMzHIa4IyV5JkC6zkY 6834
optax/contrib/_momo.py sha256=baefmkFoCcIF0cIpdkD4GnnyKJFPJGe9mrAooR-HKnY 12243
optax/contrib/_privacy.py sha256=4oOCPSTL6ItshVUr6k-vrQ06nMVEHcS87Aeow2T-qP0 4618
optax/contrib/_privacy_test.py sha256=vbSuyLm16jWXJlSkoJ9_NzEBawPkAmLQrhjuj3weCuc 4415
optax/contrib/_prodigy.py sha256=9lql1cw7s3cWVFAZUrtSDzctZL2CK3DROlA200uJGns 6033
optax/contrib/_reduce_on_plateau.py sha256=LYLSy0a_LfHvKl0jbGb2o0_HaYh-kiDBmDzuanx0xRs 6586
optax/contrib/_reduce_on_plateau_test.py sha256=0fAMwTH3T-DdysMk0V69fMJXZ6x4NPBBzvQHM6a4IPI 6482
optax/contrib/_sam.py sha256=ZLUSTzKuVX94JcaYF_2dDpRA9zR8MFJZ_JUjhgmCyms 9885
optax/contrib/_sam_test.py sha256=O_38AQY9Vtl2S7-ozZpL3j5FQyr_WLV2k3oV3ah1Azo 6755
optax/contrib/_schedule_free.py sha256=MbtaPjUdQPQz86Rm1CpHT9S3g-K6CzcuqBrbtTanbtQ 12648
optax/contrib/_schedule_free_test.py sha256=HQR8lr1lb__lw4fJNixev3Rp5e_yYcW24MIon-niw9Y 3439
optax/losses/__init__.py sha256=pLrqMNIl7_gqJ7Ei3g9wJBS1VdwaOxi-51y13M2TQZw 2360
optax/losses/_classification.py sha256=ASsRwVObbqaevQ1pPWzkDTo8IudVHDrF45AcnJpauZw 25524
optax/losses/_classification_test.py sha256=MDo7i_U69mr1JBOTimCev6BX0Is4IRKKWDx5ufhYYpo 28129
optax/losses/_fenchel_young.py sha256=c7KvlF-U-OlUR6ztRlEchmGjf0VDMAnPH-MRtMhuSx0 2076
optax/losses/_fenchel_young_test.py sha256=uxV7K0VBsKX4bQEOdLJWgvkM9tu8EgM3387Bg-0MTSg 2613
optax/losses/_ranking.py sha256=eyl-awL04r2scd4YKV_WFlGUhiQ-4iu_a5jcnkc-5VU 6204
optax/losses/_ranking_test.py sha256=6hv4MY-OcXaxTgbTEca5V-QkNJBbqhit1GOExHIX3bk 9023
optax/losses/_regression.py sha256=_wllU-M7un6MgIJP6wS582D69gM8cZ9QhufKHuSdPEA 7215
optax/losses/_regression_test.py sha256=yKn3DIOWdJIvP-Nt_dkndwz03y-pcA7DhjUW3W9jqiE 5755
optax/losses/_self_supervised.py sha256=VkyI98e6bob9-hdQyk4Ql9o8aUpUy1RgEMBirA3Ua-w 3177
optax/losses/_self_supervised_test.py sha256=fXR0s2cBctd-AmVGqiBje9V7SzVBH2Vh2UmIipONqiQ 1697
optax/losses/_smoothing.py sha256=ImSQUlkg1Q0Sxsu5FHMt1xJh6yD9-2OECtnRol13Muw 1462
optax/losses/_smoothing_test.py sha256=XbGT58HOgn1FjKIu6ovVnRYGq3-EY_KGEQiDIXZmYGQ 2331
optax/monte_carlo/__init__.py sha256=gzpzg-V1eFTxFVNDP8eljo_T8pmNR8bfhQfB9X7-jp0 1221
optax/monte_carlo/control_variates.py sha256=vud92B2kzo3TmdzGxbZmMRprNso-jn9W0fHtsy4fikA 17439
optax/monte_carlo/control_variates_test.py sha256=n1fQSfVEN5dtuSOk2aw54KUJhf2E1ZxeL8V1Xxs7MyQ 21828
optax/monte_carlo/stochastic_gradient_estimators.py sha256=i-1F8dnVSAx4ir0EO9MqGnvsGggnfJNmDfFAFYidfkU 12277
optax/monte_carlo/stochastic_gradient_estimators_test.py sha256=o3HSOXHFhoF2SKKZV5eFNn9O9l2h3mbpJaYyMEtq1tQ 13773
optax/projections/__init__.py sha256=pM5K38sXG9WhqCDoHbeXVKmz0tGvSHj-Ewr5lVYEgFo 984
optax/projections/_projections.py sha256=BCWq9y-6X-CkhsTEXvP1aWeY1uBfFGyXdbe34nQHi94 4717
optax/projections/_projections_test.py sha256=KSR02Ix7ZvwKLjMDThiThHS_tHEM1L7sv410vxoGqgk 6397
optax/schedules/__init__.py sha256=IcKlqoZs-GXX8b6LhG64vesCrGfqAPhYKRj8evj1ihw 1964
optax/schedules/_inject.py sha256=CE60mSJ3oRLURKZ5YRs3wtMZgAItUlb9tIvZQ32ANp8 10358
optax/schedules/_inject_test.py sha256=QQ7koUj_jarmJ7hpRCxXI_gQgneVMi_yNfikAGazR0E 9672
optax/schedules/_join.py sha256=lP9IdnWqULjtZuRfGnjChMPE3CQZ9nwK6recIVHdX2Q 1646
optax/schedules/_join_test.py sha256=laihh_qAtLSpCrJAPUkg6N0jc7B7WqdLWORegLE1sxI 1619
optax/schedules/_schedule.py sha256=jRWaif3JmcSX2SjTmBfin17nU42aAvLPubGZz__NxME 21087
optax/schedules/_schedule_test.py sha256=mp1EY3obke4X0iYItCmfYafmqxj-HzDqEwM-TKDcgfA 19807
optax/schedules/inject.py sha256=r-Zzd_oyV19qOr4m4jfePuI5y8Ih72_tpjo1zjyQ6Xc 897
optax/second_order/__init__.py sha256=GgYSPehA3MVDpL8U7P_CeRcsChKIm3kVwvh68AQFr0k 894
optax/second_order/_base.py sha256=b4VB6QTqMwKzVgfoVFaAqbD6cew08SQGaUKHjlTeofc 999
optax/second_order/_fisher.py sha256=MYlKA52m-8CCG5oYdHCBSqCd03NSUuRwmrNK5ts5c5A 1946
optax/second_order/_hessian.py sha256=kSo0ZO7xQ2yxno0IIkzvpYnmc7i4o-4zy5hrkG9mGD0 2529
optax/second_order/_hessian_test.py sha256=dSQB-xWV7G5Se0i8znHWBIz4kVUhspPkpwaOl6Ed--Q 2686
optax/transforms/__init__.py sha256=e2h2QlRorLr7r2Fklp2ZcMxzJ9HegsoqA818nwvNdso 3681
optax/transforms/_accumulation.py sha256=Mc_xA3oR3lCCo71v5kzQEH32iVBWml0exSRaVJrTO-Y 14683
optax/transforms/_accumulation_test.py sha256=xj625qnnzAXvYnsJQpPEdR63h3sjy4_8eTPRv5fJU2w 11721
optax/transforms/_adding.py sha256=-Qbs4jHqM3EGuvNdukQD_NPLVQeW0NuVM1AkCBnwneo 3570
optax/transforms/_adding_test.py sha256=LtdG46JgIKp7fLFnfDD7BJsQt0lapSlF4C7QOaqNmGw 3355
optax/transforms/_clipping.py sha256=sRECa32K08dSeayzsel-H2UB4PFqtfJz3lSd4cMC9Ws 10134
optax/transforms/_clipping_test.py sha256=6jxNjj07Y41C14ZEk2-N2Qs6hvY7lbTutoiLmegqsjQ 6052
optax/transforms/_combining.py sha256=T4EebzFa1CbThbIjdFiVGEF8f9igzoYXJ1A-2J-AsOo 10124
optax/transforms/_combining_test.py sha256=Xv5UJEKC36cb1sV5G4qpvubgHIiczD5JD1A9BH6UlGg 9364
optax/transforms/_conditionality.py sha256=SW_wk7PUVX423Ff3rfYbhJpPOY0Yft64W72Ddu6wl-g 9305
optax/transforms/_conditionality_test.py sha256=3sawn8kdebNR4r-2WhfBbN6TS10sPdWc-6fJNjxyWMM 10639
optax/transforms/_constraining.py sha256=KLnQFbJ3993RmOUYDJJSjeGlfL5kgJNQPDA2gBRX3YE 3135
optax/transforms/_constraining_test.py sha256=Ub0p7QXwWXNB4frzKl9JdEGhUyR2fl55FyKQfCyiYoI 4311
optax/transforms/_layouts.py sha256=b64uXAUIguTrVmsyCB5etDMhdxXQo5RKRENsBPxL8i0 2627
optax/transforms/_layouts_test.py sha256=KIiajIPPRuap9q6BjzzJj_HuHVhERvKMbrDqoYL9CM4 1995
optax/transforms/_masking.py sha256=B9LPz7y4ubyUAqxEruilznkn_5UmYzhF9R58xkt8YpE 5939
optax/transforms/_masking_test.py sha256=7iW59KmSPaudZVNyerMhkgKpVH82n_5IK0ZQyrK0dnk 12056
optax/tree_utils/__init__.py sha256=sqiEI7ShBqm3WsolPHzzFIr94rVEo6EzZ_meWNC8GjI 2196
optax/tree_utils/_casting.py sha256=4wyyDnvzoG_kP0XI9ltR8YbGiAcwv4pxnkdGzg8XtNY 1077
optax/tree_utils/_casting_test.py sha256=42EoLu9pu1vCCuYFD-PAuYP-0BFx0UMOLU0yN-PUs-A 1812
optax/tree_utils/_random.py sha256=pNjTVDiFvzAMt40ZTrw5cpkuKd8qiYoeAbXsr7LO9CM 2091
optax/tree_utils/_random_test.py sha256=F0hpq5GL2t_xffMcs_uOrQ8TS0dlngSP5J90_HQfPeA 3647
optax/tree_utils/_state_utils.py sha256=6eadR5cI5eRHQ5z_uYmy_YVE2TQGxyP8brov9j8OwGI 27890
optax/tree_utils/_state_utils_test.py sha256=dPLzfsSSaXHGYK-5srDgCHXJA2ehsS9ZrA_3C23Qorg 20342
optax/tree_utils/_tree_math.py sha256=nDLWFwrewSR9lmYr33dmBv-VcyP5IL_31Oypy-Ifpiw 9002
optax/tree_utils/_tree_math_test.py sha256=1mUB3poYNZwycGbg4x3bn4t23aNFYrApi8fEhZJP6HI 8508
optax-0.2.3.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
optax-0.2.3.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
optax-0.2.3.dist-info/METADATA sha256=nT7NncsNbuD6zj22jDmt9Rnsu7c3cniFnWbZjVM-V7A 8287
optax-0.2.3.dist-info/RECORD — —