Skip to main content

Ctrl+K

Examples

Getting Started
Tutorials

API

ott.geometry
ott.problems
- ott.problems.linear
- ott.problems.quadratic
ott.solvers
ott.initializers
- ott.initializers.linear
- ott.initializers.quadratic
  - ott.initializers.quadratic.initializers.QuadraticInitializer
ott.neural
ott.tools
ott.math
ott.utils
- ott.utils.default_progress_fn
- ott.utils.tqdm_progress_fn

References

Bibliography
Contributing Guide

.rst

ott.neural.networks.potentials.PotentialTrainState.tx

Contents

PotentialTrainState.tx

ott.neural.networks.potentials.PotentialTrainState.tx#

PotentialTrainState.tx: optax.GradientTransformation#

previous

ott.neural.networks.potentials.PotentialTrainState.params

next

ott.neural.networks.potentials.PotentialTrainState.opt_state

Contents

PotentialTrainState.tx

By OTT authors

© Copyright 2021-2024, OTT authors.