Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX
wesselb
Back