tidy3d.plugins.autograd.optimizers.adam

Contents

tidy3d.plugins.autograd.optimizers.adam#

class adam[source]#

Bases:

Create an Adam optimizer (convenience factory, mirrors optax.adam).

Parameters:
  • learning_rate (float) – Step size for the parameter updates.

  • beta1 (float = 0.9) – Exponential decay rate for the first moment estimate.

  • beta2 (float = 0.999) – Exponential decay rate for the second moment estimate.

  • eps (float = 1e-8) – Small constant for numerical stability.

Returns:

Configured Adam optimizer instance.

Return type:

Adam