Ecosyste.ms: Repos

An open API service providing repository metadata for many open source software ecosystems.

GitHub / google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

JSON API: https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google-deepmind%2Foptax

Stars: 1,505
Forks: 154
Open Issues: 60

License: apache-2.0
Language: Python
Repo Size: 4.15 MB
Dependencies: 36

Created: almost 4 years ago
Updated: 3 days ago
Last pushed: 3 days ago
Last synced: 3 days ago

Topics: machine-learning, optimization

Files

Loading...

Readme

Loading...

Dependencies

.github/workflows/pypi-publish.yml actions

actions/checkout v2 composite
actions/setup-python v4 composite

.github/workflows/tests.yml actions

actions/checkout v2 composite
actions/setup-python v4 composite

pyproject.toml pypi

absl-py >=0.7.1
chex >=0.1.5
jax >=0.1.55
jaxlib >=0.1.37
numpy >=1.18.0

requirements/minimum-requirements-dp-accounting.txt pypi

absl-py >=1.0.0
attrs >=21.4.0
mpmath >=1.2.1
numpy >=1.21.4
scipy >=1.7.1

requirements/requirements-docs.txt pypi

IPython ==7.16.3
dm-haiku ==0.0.8
docutils ==0.16
ipykernel ==5.3.4
matplotlib ==3.5.0
myst_nb ==0.13.1
pandoc ==1.0.2
sphinx ==4.5.0
sphinx-autodoc-typehints ==1.11.1
sphinx-book-theme ==0.3.3
sphinxcontrib-bibtex ==2.4.2
sphinxcontrib-katex ==0.9.0

requirements/requirements-examples.txt pypi

dm-haiku >=0.0.3
tensorflow >=2.4.0
tensorflow-datasets >=4.2.0

requirements/requirements-test.txt pypi

dm-haiku >=0.0.3 test
dm-tree >=0.1.7 test

requirements/requirements.txt pypi

absl-py >=0.7.1
chex >=0.1.8
jax >=0.4.13
jaxlib >=0.4.13
numpy >=1.25.0

setup.py pypi