@jax-js/optax
This is a port of Optax to jax-js, for gradient
processing and optimization. It includes implementations of common algorithms like Adam.
Interfaces
- GradientTransformation
Type Aliases
- OptState
- ScaleByAdamOptions
Functions
- adam
- applyUpdates
- identity
- l2Loss
- scaleByAdam
- setToZero
- sgd
- squaredError