Piecewise Linear Functions (PWL) implementation in PyTorch - View it on GitHub