GitHub repository: lucidrains/x-unet
Primary language: Python
Description: Implementation of a U-net complete with efficient attention as well as the latest research findings