lucidrains / lion-pytorchView external linksLinks
🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
☆2,184Nov 27, 2024Updated last year
Alternatives and similar repositories for lion-pytorch
Users that are interested in lion-pytorch are comparing it to the libraries listed below
Sorting:
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,336Feb 5, 2026Updated last week
- Accessible large language models via k-bit quantization for PyTorch.☆7,952Updated this week
- Fast and memory-efficient exact attention☆22,231Updated this week
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆981Jan 30, 2024Updated 2 years ago
- Schedule-Free Optimization in PyTorch☆2,256May 21, 2025Updated 8 months ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,491Feb 6, 2026Updated last week
- An open source implementation of CLIP.☆13,383Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,395Jan 26, 2026Updated 3 weeks ago
- Vector (and Scalar) Quantization, in Pytorch☆3,870Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,162Mar 22, 2024Updated last year
- D-Adaptation for SGD, Adam and AdaGrad☆528Jan 22, 2025Updated last year
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,021Jan 23, 2026Updated 3 weeks ago
- Implementation of GigaGAN, new SOTA GAN out of Adobe. Culmination of nearly a decade of research into GANs☆1,943Jan 12, 2025Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆641Dec 19, 2025Updated last month
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,351Updated this week
- Foundation Architecture for (M)LLMs☆3,130Apr 11, 2024Updated last year
- Code release for ConvNeXt model☆6,292Jan 8, 2023Updated 3 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,351Updated this week
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆24,993Updated this week
- Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"☆8,352May 31, 2024Updated last year
- Code release for ConvNeXt V2 model☆1,964Aug 14, 2024Updated last year
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,823Feb 4, 2026Updated last week
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆722Oct 16, 2023Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆807Jun 8, 2025Updated 8 months ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,384Oct 28, 2024Updated last year
- Implementation of Denoising Diffusion Probabilistic Model in Pytorch☆10,455Aug 4, 2025Updated 6 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆292May 3, 2024Updated last year
- Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.☆4,358Oct 19, 2025Updated 3 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆804Jan 30, 2026Updated 2 weeks ago
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆20,619Feb 9, 2026Updated last week
- Karras et al. (2022) diffusion models for PyTorch☆2,561Jan 7, 2025Updated last year
- LAVIS - A One-stop Library for Language-Vision Intelligence☆11,166Nov 18, 2024Updated last year
- EVA Series: Visual Representation Fantasies from BAAI☆2,648Aug 1, 2024Updated last year
- PyTorch code and models for the DINOv2 self-supervised learning method.☆12,393Dec 22, 2025Updated last month
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,248Updated this week
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,990Feb 9, 2026Updated last week
- 🐍 Geometric Computer Vision Library for Spatial AI☆11,071Feb 9, 2026Updated last week
- 🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.☆32,768Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆7,077Updated this week