Thank you for the insight. I recently encountered this paper 6 and it’s implementation in TF 8. Although they address a more general case where the constraints are not differentiable using a so called proxy Lagrangian, I was wondering if you have encountered similar PyTorch implementations (Not necessarily for non-differentiable constraints, but only general enough to work for non-convex DNNs).