flax
Add test case for PixelCNN to run a single training and evaluation step.
#654
Merged
Merged