optimal-kernels Optimal convolution kernels project Datasets /dataset_npy - simple shapes dataset Models /unet_basic/pytorch_unet.py - baseline U-Net implemented in PyTorch