PyTorch library for "Neural Painters: A learned differentiable constraint for generating brushstroke paintings"