A PyTorch Framework for Learning with Constraints

Pylon lets you train your deep learning models with arbitrary constraints on the output.

Github Video

Easy to Write logo

Easy to Write

Single line of code that captures the constraint.
Integrates with PyTorch logo

Integrates with PyTorch

Do not have to change existing code, just add a new loss.
Efficient Training logo

Efficient Training

Compile the constraints into losses that are optimized efficiently.