Getting Started
Tutorials
Reference Documentation
Extending NetKet
API documentation
Developer Documentation
Adam Optimizer.
learning_rate (float) – Learning rate \(\eta\).
float
b1 (float) – Decay rate for the exponentially weighted average of grads.
b2 (float) – Decay rate for the exponentially weighted average of squared norm of grads.
eps – Term added to the denominator to improve numerical stability.