Implementation of a U-net complete with efficient attention as well as the latest research findings
☆292May 3, 2024Updated last year
Alternatives and similar repositories for x-unet
Users that are interested in x-unet are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Feb 13, 2023Updated 3 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,183Nov 27, 2024Updated last year
- Optimized library for large-scale extraction of frames and audio from video.☆201Sep 11, 2023Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Jul 9, 2023Updated 2 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆70Dec 24, 2022Updated 3 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆207Aug 26, 2023Updated 2 years ago
- A fully invertible U-Net for memory efficiency in Pytorch.☆130Jul 9, 2022Updated 3 years ago
- Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch☆879Oct 30, 2023Updated 2 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆135Oct 15, 2025Updated 5 months ago
- Official implementation of Cold-Diffusion for different transformations in pytorch.☆1,120Oct 13, 2022Updated 3 years ago
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 3 years ago
- (unofficial) - customized fork of DETR, optimized for intelligent obj detection on 'real world' custom datasets☆12Aug 22, 2020Updated 5 years ago
- Axial Positional Embedding for Pytorch☆84Feb 25, 2025Updated last year
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Nov 26, 2022Updated 3 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆804Jan 30, 2026Updated last month
- Karras et al. (2022) diffusion models for PyTorch☆14Nov 15, 2022Updated 3 years ago
- A C++/Cython audio limiter for Python.☆25Feb 8, 2023Updated 3 years ago
- Implementation of the convolutional module from the Conformer paper, for use in Transformers☆433May 17, 2023Updated 2 years ago
- Algorithms to automatically recognize guitar effects and retrieve their parameters for timbre reproduction☆25Aug 31, 2022Updated 3 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆196Mar 27, 2021Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Dec 21, 2021Updated 4 years ago
- Directed masked autoencoders☆14Mar 17, 2026Updated last week
- A Pytorch implementation of Stylegan2 with UNet Discriminator☆216Nov 10, 2020Updated 5 years ago
- Implementation of a Transformer, but completely in Triton☆279Apr 5, 2022Updated 3 years ago
- PyTorch reimplementation of Noise2Same with enhancements☆11Mar 6, 2026Updated 2 weeks ago
- Implementation of N-Grammer in Flax☆17Nov 3, 2022Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Dec 4, 2021Updated 4 years ago
- Implementation of a holodeck, written in Pytorch☆18Nov 1, 2023Updated 2 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,986Jun 16, 2024Updated last year
- Tutorial covering Open Source tools for Source Separation.☆15Nov 12, 2021Updated 4 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,438Feb 20, 2026Updated last month
- v objective diffusion inference code for PyTorch.☆718Nov 29, 2022Updated 3 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,419Updated this week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,752Jan 20, 2026Updated 2 months ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Oct 22, 2023Updated 2 years ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 3 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆518Sep 28, 2023Updated 2 years ago
- Karras et al. (2022) diffusion models for PyTorch☆2,575Feb 12, 2026Updated last month
- A python implementation of a traditional Dynamic Range Compressor☆14Oct 30, 2020Updated 5 years ago