Jed Homer

Thanks for your reply and the recommendations! The main thing I am looking to do is a Haar 3D-DWT as part of a parallelised layer in `Pytorch`. I think all...

No I didn't - it's an area of mystery to me but I would be very interested in a discussion / looking into a solution! There doesn't seem to be...

Following @rosinality's advice, it works with label conditioning on MNIST. I added a label conditioning to the affine coupling layers instead of the prior (this is good because the net...

Thanks @patrick-kidger, I will definitely use `equinox` next time around, its design is incredible.