Implementation of UltraMem, improved Product Key Memory design, from Bytedance AI labs
☆28Nov 4, 2025Updated last year
Alternatives and similar repositories for ultra-mem
Users that are interested in ultra-mem 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 Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated last year
- Embedding and readout for simple multi-categorical and gaussian continuous☆20Jul 5, 2026Updated 4 months ago
- Unofficial implementation of Hippoformer, Integrating Hippocampus-inspired Spatial Memory with Transformers☆57Aug 4, 2026Updated 3 months ago
- Some utility functions to help myself (and perhaps others) go faster with ML/AI work☆55Updated this week
- Just a repository that will house some MLPs and their variants, so to avoid having to reimplement them again and again for different proj…☆51Updated this week
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Unofficial implementation of Tiny Recursive Model (TRM), improvement to HRM from Sapient AI, by Alexia Jolicoeur-Martineau☆192Dec 23, 2025Updated 10 months ago
- Implementation of various evolutionary algorithms, starting with evolutionary strategies☆53Updated this week
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆17May 19, 2023Updated 3 years ago
- Implementation and explorations into PopuLoRA, Co-Evolving LLM Populations for Reasoning Self-Play☆19Oct 17, 2026Updated 2 weeks ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆112Oct 27, 2026Updated last week
- Code for Fast-weight Product Key Memory (FwPKM)☆27Mar 18, 2026Updated 7 months ago
- a fast and lightweight distributed background task processing framework with seamless scheduling.☆15Mar 30, 2026Updated 7 months ago
- Run ops on Apple ANE in NPU register with pure python on M1 Asahi Linux. No Espresso, No CoreML, no metal, no .mlmodels file, no .hwx fil…☆21Oct 8, 2026Updated 3 weeks ago
- ☆20Sep 16, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Implementation of Poly-attention, a higher-order self-attention proposed by Chakrabarti et al. of Columbia☆55Aug 19, 2026Updated 2 months ago
- 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
- Implementation of the fast weight product key memory from Sakana AI☆21Sep 17, 2026Updated last month
- a simple API to use CUPTI☆10Aug 19, 2025Updated last year
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆28Oct 7, 2026Updated 3 weeks ago
- Implementation of Multiscreen proposed by Ken Nakanishi for "Screening is Enough"☆19May 13, 2026Updated 5 months ago
- Conformer block with Rotary Position Embedding, modified from lucidrains' implement☆19Sep 13, 2024Updated 2 years ago
- Code for the paper "Cottention: Linear Transformers With Cosine Attention"☆21Nov 15, 2025Updated 11 months ago
- ☆11May 2, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Reduce kernel based on CUTLASS CuTe and TMA.☆12Sep 25, 2025Updated last year
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆145Sep 3, 2026Updated 2 months ago
- Atomistic Generative Diffusion☆15Sep 1, 2026Updated 2 months ago
- ☆13Oct 29, 2024Updated 2 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆51Oct 9, 2026Updated 3 weeks ago
- Learning to Skip the Middle Layers of Transformers☆17Aug 7, 2025Updated last year
- Official repo for BWLer: Barycentric Weight Layer☆34Aug 11, 2026Updated 2 months ago
- Source code for paper Fused3S: Fast Sparse Attention on Tensor Cores☆11Oct 13, 2025Updated last year
- Official code for "Outcome-directed Reinforcement Learning by Uncertainty & Temporal Distance-Aware Curriculum Goal Generation" (ICLR 202…☆12Jan 30, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A Machine Learning Benchmark for Predicting Enhanced-Precision Electronic Charge Density in Crystalline Inorganic Materials☆15Feb 9, 2025Updated last year
- Jax Codebase for Evolutionary Strategies at the Hyperscale☆385Feb 27, 2026Updated 8 months ago
- Open source repo for simple SOTA method.☆14Apr 21, 2026Updated 6 months ago
- ☆13Jul 21, 2026Updated 3 months ago
- Reproducibility study of the paper: "InPars Toolkit: A Unified and Reproducible Synthetic Data Generation Pipeline for Neural Information…☆12Feb 11, 2025Updated last year
- A set of custom nodes that I've either written myself or adapted from other authors for my own convenience.☆11Sep 18, 2024Updated 2 years ago
- BEAVER: A Training-Free Hierarchical Prompt Compression Method via Structure-Aware Page Selection☆18Mar 30, 2026Updated 7 months ago