Definitely! It might sound like a sales-thing to say, but PyTorch in general is a huge point on our roadmap, but we’re not 100% sure what the most logical approach is. To keep the APIs consistent will involve some fiddling on our end.