Wheelodex » Projects » optax

optax

View on PyPI — Reverse Dependencies (209)

0.2.2 optax-0.2.2-py3-none-any.whl

Wheel Details

Project: optax
Version: 0.2.2
Filename: optax-0.2.2-py3-none-any.whl
Download: [link]
Size: 223674
MD5: 711cf28deab970cce394dc9b976c51e1
SHA256: 411c414a76aae259f4191a60b712663968741a5163ca92fc250b5d5c7d36fb57
Uploaded: 2024-03-27 18:05:53 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.1
Name: optax
Version: 0.2.2
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.1.55)
Requires-Dist: jaxlib (>=0.1.37)
Requires-Dist: numpy (>=1.18.0)
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: flax; extra == "examples"
Requires-Dist: dm-tree (>=0.1.7); extra == "test"
Requires-Dist: flax (>=0.5.3); extra == "test"
Provides-Extra: docs
Provides-Extra: dp-accounting
Provides-Extra: examples
Provides-Extra: test
Description-Content-Type: text/markdown
[Description omitted; length: 5593 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=XdnxTTdfTaJe8rJyusrLIv4tPPj_g8I0G5HAqSOyDbw 14054
optax/optax_test.py sha256=nagsZ9x8iSG4ZFd8LroMNpi2gMq_OIhj4egIPr5WVP0 986
optax/_src/alias.py sha256=pWEMq82cDaGLRcHOxPtgH7NQb663WrCt5UTO-eOJhZ8 72030
optax/_src/alias_test.py sha256=br7bSeBKuEajXqeowVINdkzaWmk06k16w3bolOXfNhk 7879
optax/_src/base.py sha256=lzBF6Q0bcT1N0aH28JhAEryUTHJzk34kI8WEf2majNc 12042
optax/_src/base_test.py sha256=z-cv8FVJrtUVEKvFdA4oInCGxXdzUK4rXFsI_L3zO84 6609
optax/_src/clipping.py sha256=QukDLsm7aps4AwU60Srqo-dPxxKe2KIlhYW8_rqdkDE 10492
optax/_src/clipping_test.py sha256=LKZPmVgBS2QQneVL7gzFqdSduf8PEP8KxPLkfkzxRDY 6024
optax/_src/combine.py sha256=_Aw83eYCuECgqfYyo52nt3wTaMCSJHqt0gQdxiQZh8g 9650
optax/_src/combine_test.py sha256=APahLo52tFywrpw1SX4BBnNNDODlfN0Sg3pQKeY2mTw 9351
optax/_src/constrain.py sha256=6Akhu8iwsfzQxnevbjby_lv_ArX1CFhyv6BvBUGOJWc 3150
optax/_src/constrain_test.py sha256=2uKf9ETc0gDkehjIH_Euev4KrhLwLOQDUlZx2xWyZQo 4200
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=joFiJBoSRTvVawMhdQtIWr5XA_hfQPPh0aCj397ujGA 7225
optax/_src/linesearch.py sha256=LbmeRB7udCUCuRnJ5ISlBDGghJSx5dwsOavtMcYM2J0 14953
optax/_src/linesearch_test.py sha256=rTOWDIuFPXiXIk8OOFIXAgskujqf7i4pzk02FEJglkU 8109
optax/_src/lookahead.py sha256=BXerdCUxokCE-yeLZWMfgXhHLoXvqQrc0Dbl_9HSqw4 7471
optax/_src/lookahead_test.py sha256=qRot9uAOJnB1sEqF7SO9lCOF4mLTLizdYsF8J3UZ-jc 5713
optax/_src/numerics.py sha256=atUjLyfVFNfmMC-Xf0gGxxsGDgEgssrwA-yvPMp4A0I 4636
optax/_src/numerics_test.py sha256=YRMQQU63pjkIuRmuMF5Vf3Wk9AGqcoM7ObExGVXsPuM 4198
optax/_src/schedule.py sha256=JVnrKRsErCVHwO2P-_KRuCG3H7J5lBtzC6CLyz2_c6E 1905
optax/_src/transform.py sha256=E5J4HWeZq144FQJP1Gk07niYP7HL-19luUFd5-dAmiw 48677
optax/_src/transform_test.py sha256=Lj4i3oJdwxw7tu4-DaxEPyDoAW3Kb0DPWYva_PVQxs4 11612
optax/_src/update.py sha256=kYH4w9lBTY1kS3G6IpLpteIF-gLcvimk6kdH3ZxJZz4 3692
optax/_src/update_test.py sha256=23TBGtiYf8SCy0EzGobExY14AmbVXU3mF4VJlp7Xb5c 2874
optax/_src/utils.py sha256=1uJUj_gXiA_SpVvIsA09SvF38Uj_JITQG27QsJ4IxkQ 10592
optax/_src/utils_test.py sha256=p3vefxdzqjptEzUHzuEBWIOnTBB_y9rQLrZMGv6OGxo 10756
optax/_src/wrappers.py sha256=ltzq-6NyODwVPyJ2qER-1j4dy-0284xlwB9qITUN75Q 24079
optax/_src/wrappers_test.py sha256=fYEx_jVZeQ2U8lbzfVuv35fV0ih6X8qejWL2q2eYC-4 29589
optax/contrib/__init__.py sha256=VS8zwwi6uq_37MQn77GL5eYvyLZ3CDGjjMhIOJATbYI 1725
optax/contrib/_cocob.py sha256=ZY07nG7IhGL0o4JxbntQdcj55w3UIpDB0MFf-4TarPE 3423
optax/contrib/_common_test.py sha256=JEA9O0c87mLAodtatu36lR-LuquE02UH06eO3pGZ7iY 4261
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/_mechanic.py sha256=OiDSMyinqu-xN01kLTw3Rg9mIe5JZmfiXyk5P_0gndY 7376
optax/contrib/_mechanic_test.py sha256=mOK3HLc2cZ3bbzLTm17dQWzH91Gwm-qzEoH-CpQNGxg 6834
optax/contrib/_privacy.py sha256=4oOCPSTL6ItshVUr6k-vrQ06nMVEHcS87Aeow2T-qP0 4618
optax/contrib/_privacy_test.py sha256=vbSuyLm16jWXJlSkoJ9_NzEBawPkAmLQrhjuj3weCuc 4415
optax/contrib/_prodigy.py sha256=6lTLQ-YGaLcnU0OD5tYIwhgo3DRIHzYYQpnOSnFXzf8 5711
optax/contrib/_reduce_on_plateau.py sha256=YYP8c8rEeO0tSA5mpFHDzU_bCCvGLmNwtYSX60F4q6U 6242
optax/contrib/_reduce_on_plateau_test.py sha256=hVHnZxWkU_mLEQV6RntF_a1BxnREhcEKP-EbXPU_Irs 5089
optax/contrib/_sam.py sha256=ri8ygx6KqBybELp5tQ17UeVQgUPY2HYH9yeHDA5ge9c 9857
optax/contrib/_sam_test.py sha256=Xo0BLg4LslQYV5W03Lt1A76lHOy8v5ptbhlGH7Q1cVg 6747
optax/losses/__init__.py sha256=kSX-La4RtSenurCNNhE1ZP5guGcX_EhsPEkY5tusq8I 1895
optax/losses/_classification.py sha256=Bn3B3NZi28PYl6G8oB8cqSrAPzkObfq3PBiXkC6eDI4 21052
optax/losses/_classification_test.py sha256=1rGg2R-cjpY54fPcJ4tBQlqwoLslncmsbegDSPmGxrc 21622
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/_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=PZ5Sl9TkHIPcZrnsSXS_7CIyJGXV9GmJlliWgEQHGKM 17499
optax/monte_carlo/control_variates_test.py sha256=Sdw7saJd7luRJi4tXeNR52FGfNrvuy6_2eSnqzLgxzE 21803
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=mHN5YLSdl6QMg9JvflANa-kD4yL6IaaUsxLb_2ngKgY 922
optax/projections/_projections.py sha256=G9qNiX8e5_FJMThYeDDSneUk-k1IfjJGLuK7XaF07ic 2628
optax/projections/_projections_test.py sha256=O9y6nvI8AvTIANbIaP706uqgF9sPe_nCek4QNlyAFbE 3932
optax/schedules/__init__.py sha256=rrW-DIHEblTF3MkLPF_GbVh7Q2FM7dsLlhV0wIheHWY 1901
optax/schedules/_inject.py sha256=5wevQbXurRH4BNw94nDzrSjHrOFYYhuRXFVGXu8CL_Q 10318
optax/schedules/_inject_test.py sha256=SIZpjppg9PTLcUcua2pYgmPUXjtFMR50DqgfuavOajA 9671
optax/schedules/_join.py sha256=lP9IdnWqULjtZuRfGnjChMPE3CQZ9nwK6recIVHdX2Q 1646
optax/schedules/_join_test.py sha256=laihh_qAtLSpCrJAPUkg6N0jc7B7WqdLWORegLE1sxI 1619
optax/schedules/_schedule.py sha256=a7QKjxpmWgTvwE6D_Taaf5DpUVxhwUG5rZiuo035l1U 20488
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/tree_utils/__init__.py sha256=UZeEJNSCLQ0hpAc0X8XJchDik2KKh3MAWzQyui2VJxY 1646
optax/tree_utils/_state_utils.py sha256=cbaLHzw9tbFE1j-zIO2OLCCADsiUeEHBrR4ZjLRO72g 27870
optax/tree_utils/_state_utils_test.py sha256=9EsKEk1clCRD57KsEXX8g5O8elP-i57bgHapIPGrhCI 20342
optax/tree_utils/_tree_math.py sha256=h-ninvzoD-Z388XQV5NeYlEfOlAvuiNtJWB8dJrqQPw 6159
optax/tree_utils/_tree_math_test.py sha256=PD3VWaX3S9YDOtwYJ0e31kGGW91qRnKtGU1-CMU1K9w 7734
optax-0.2.2.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
optax-0.2.2.dist-info/WHEEL sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4 81
optax-0.2.2.dist-info/METADATA sha256=efqifvn-EbRIsRPiaWR6EHL9969iS84BkbhZTSrbGvk 8098
optax-0.2.2.dist-info/RECORD — —