LoreonLabsPlatform

Overview

Intelligence

Markets
Builders
Research
Narratives
Ecosystems
Launchpads

Discover

Search
Sources

Python

optax

Optax is a gradient processing and optimization library for JAX.

PythonEmerging

Stars

—

Forks

—

Contributors

8

Last push

3d ago

Recent commits

Latest commits.

Ignore optax badge link in link check
d890640Robert Dyro3d ago
Merge pull request #1698 from rajasekharporeddy:flax_example
879e093OptaxDev3d ago
Merge pull request #1703 from Osamaali313:docs/cosine-similarity-epsilon
03ca703OptaxDev3d ago
Fix mlc_config.json for markdown-link-check
f41bc04Robert Dyro3d ago
Merge pull request #1699 from rajasekharporeddy:gradient_acc
4192e9dOptaxDev3d ago

docs(losses): clarify epsilon semantics in cosine_similarity/cosine_distance

5390ea4Osamaali3134d ago

Reduce focal loss memory usage.

2fe2297OptaxDev7d ago

Narrow the return type of `safe_increment` to `jax.Array`.

465819fAdam Hillier7d ago

Top contributors

Builders behind this project.