optax
A gradient processing and optimization library in JAX.
- Category
- Python packages
pip install optax Official docs outline
optax.readthedocs.ioExtra Credit builds an outline of optax's docs the first time your work uses it. Until then, start at the official documentation.
More in Python packages
All 9,553-
flatten-json Flatten JSON objects
-
mypy-boto3-iam Type annotations for boto3 IAM 1.43.70 service generated with mypy-boto3-builder 8.12.0
-
ast-grep-cli Structural Search and Rewrite code at large scale using precise AST pattern.
-
model-hosting-container-standards Python toolkit for standardized model hosting container implementations with Amazon SageMaker integration
-
hijri-converter [DEPRECATED] Use 'hijridate' package instead
-
zipfile-zstd Monkey patch the standard zipfile module to enable Zstandard support
-
gspread-dataframe Read/write gspread worksheets using pandas DataFrames
-
evaluate HuggingFace community-driven open-source library of evaluation
-
suds-community Lightweight SOAP client (community fork)
-
doclang DocLang reference toolkit
-
inspect-ai Framework for large language model evaluations
-
gensim Python framework for fast Vector Space Modelling