Schedule-Free Optimization in PyTorch
☆2,330Jul 28, 2026Updated 3 months ago
Alternatives and similar repositories for schedule_free
Users that are interested in schedule_free are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Efficient optimizers☆340Sep 9, 2026Updated last month
- A PyTorch native platform for training generative AI models☆5,858Updated this week
- TensorDict is a pytorch dedicated tensor container.☆1,043Updated this week
- For optimization algorithm research and development.☆583Oct 23, 2026Updated last week
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆979Nov 16, 2025Updated 11 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Efficient Triton Kernels for LLM Training☆6,678Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆322Jun 11, 2024Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,919May 24, 2026Updated 5 months ago
- Tile primitives for speedy kernels☆3,786Sep 12, 2026Updated last month
- ☆285Dec 2, 2024Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆605Aug 14, 2026Updated 2 months ago
- maximal update parametrization (µP)☆1,756Jul 17, 2024Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,627Aug 26, 2026Updated 2 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,206Jul 9, 2026Updated 3 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.
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,772Jul 18, 2025Updated last year
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆462May 13, 2025Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,899Jun 22, 2025Updated last year
- NanoGPT (124M) in 90 seconds☆5,999Sep 28, 2026Updated last month
- PyTorch native post-training library☆5,821Sep 9, 2026Updated last month
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,266Aug 22, 2025Updated last year
- PyTorch native quantization for training and inference☆3,018Updated this week
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection☆1,710Oct 28, 2024Updated 2 years ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,572Updated this week
- 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.
- Helpful tools and examples for working with flex-attention☆1,262Updated this week
- Mamba SSM architecture☆18,960Jul 22, 2026Updated 3 months ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆1,000Jan 30, 2024Updated 2 years ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,604Jan 12, 2025Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,335Aug 26, 2025Updated last year
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,475Oct 12, 2026Updated 2 weeks ago
- Fast and memory-efficient exact attention☆25,221Updated this week
- 🧱 Modula software package☆347Aug 18, 2025Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆437Dec 12, 2024Updated last year
- 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.
- Implementation for MatMul-free LM.☆3,100Sep 6, 2026Updated last month
- Minimalistic large language model 3D-parallelism training☆2,850Oct 8, 2026Updated 3 weeks ago
- UNet diffusion model in pure CUDA☆669Jun 28, 2024Updated 2 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,961Oct 23, 2026Updated last week
- 🚀 Efficient implementations for emerging model architectures☆6,004Updated this week
- Accessible large language models via k-bit quantization for PyTorch.☆8,535Updated this week
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,824Jan 5, 2026Updated 9 months ago