Can you elaborate? I feel like torch.nn gets you pretty close.