Some utility functions to help myself (and perhaps others) go faster with ML/AI work
☆54Oct 5, 2026Updated last week
Alternatives and similar repositories for torch-einops-utils
Users that are interested in torch-einops-utils 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 Recurrent Independent Mechanisms in Pytorch☆27Apr 6, 2026Updated 6 months ago
- Explorations into the proposed Streaming Deep Reinforcement Learning, from University of Alberta☆32Sep 6, 2026Updated last month
- Contrastive Reinforcement Learning☆73Sep 28, 2026Updated 2 weeks ago
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated last year
- Implementation of UltraMem, improved Product Key Memory design, from Bytedance AI labs☆28Nov 4, 2025Updated 11 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Explorations into NEAT and some of its derivative research☆42Sep 7, 2026Updated last month
- Exploration into some new research surrounding value networks☆16Feb 10, 2026Updated 8 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆81Sep 2, 2026Updated last month
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆17Apr 22, 2025Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆156May 2, 2025Updated last year
- The Hessian of tall-skinny networks is easy to invert☆17Sep 5, 2026Updated last month
- Implementation of a transformer for reinforcement learning using `x-transformers`☆72Sep 25, 2025Updated last year
- Implementation of the fast weight product key memory from Sakana AI☆20Sep 17, 2026Updated last month
- ImageNet-12k subset of ImageNet-21k (fall11)☆23Jun 13, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆216Sep 11, 2026Updated last month
- ☆15Dec 26, 2025Updated 9 months ago
- The full training script for Enformer - Tensorflow Sonnet☆18Apr 18, 2022Updated 4 years ago
- Official code for "ZigZag: Universal Sampling-free Uncertainty Estimation Through Two-Step Inference" (TMLR 2024)☆17Nov 7, 2024Updated last year
- Implementation and explorations into PopuLoRA, Co-Evolving LLM Populations for Reasoning Self-Play☆19Oct 8, 2026Updated last week
- Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch☆107Apr 3, 2026Updated 6 months ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆28Sep 23, 2021Updated 5 years ago
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆68Apr 7, 2025Updated last year
- Karras et al. (2022) diffusion models for PyTorch☆14Nov 15, 2022Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Efficient implementation (and explorations) into polar coordinate positional embedding (PoPE) - from Gopalakrishnan et al. under Schmidhu…☆74Jun 21, 2026Updated 3 months ago
- Sharing how attention appears in my mind's eye, for students and those interested in understanding transformers☆60Updated this week
- A CLI for controlling scanners on MacOS☆10Feb 4, 2026Updated 8 months ago
- Implementation of HS-TasNet, "Real-time Low-latency Music Source Separation using Hybrid Spectrogram-TasNet"☆113Updated this week
- ☆48Oct 31, 2025Updated 11 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆188May 13, 2026Updated 5 months ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆31Apr 11, 2024Updated 2 years ago
- Explorations into some of the approaches advocated by Yann LeCun, and just a more wholistic architecture (JEPA) in general☆126Sep 21, 2026Updated 3 weeks ago
- Embedding and readout for simple multi-categorical and gaussian continuous☆20Jul 5, 2026Updated 3 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Thinker project☆16Sep 4, 2024Updated 2 years ago
- Code for experiments to learn uncertainty☆31Mar 16, 2023Updated 3 years ago
- Tools for visualizing neural nets☆22Jul 29, 2025Updated last year
- Getting crystal-like representations with harmonic loss☆196Apr 2, 2025Updated last year
- Sci-Reasoning: A Dataset Decoding AI Innovation Patterns☆19Jan 13, 2026Updated 9 months ago
- Official Implementation of "RTop-K: Ultra-Fast Row-Wise Top-K Selection for Neural Network Acceleration on GPUs"☆30Jul 23, 2025Updated last year
- Digital Red Queen: Adversarial Program Evolution in Core War with LLMs☆226Jan 13, 2026Updated 9 months ago