Concept
Stop-gradient
Stop-gradient is an operation that lets a tensor flow forward through your computation while blocking any gradient from flowing back through it during backpropagation. detach(); in JAX/TensorFlow it is stop_gradient(tensor)`.
The rest of “Stop-gradient” is a premium feature: every concept in the library gets a precise, practitioner-focused write-up like this one, cross-linked straight from the paper summaries that use it.
Log in to unlock→