lucidrains / x-transformers

Related projects: ⓘ

• huggingface / accelerate
  🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…
  ☆7,687Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,362Updated this week
• lucidrains / lion-pytorch
  🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
  ☆2,002Updated 3 months ago
• lucidrains / denoising-diffusion-pytorch
  Implementation of Denoising Diffusion Probabilistic Model in Pytorch
  ☆7,916Updated 2 weeks ago
• state-spaces / s4
  Structured state space sequence models
  ☆2,361Updated 2 months ago
• lucidrains / vector-quantize-pytorch
  Vector (and Scalar) Quantization, in Pytorch
  ☆2,404Updated this week
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,003Updated 5 months ago
• Dao-AILab / flash-attention
  Fast and memory-efficient exact attention
  ☆13,401Updated this week
• diff-usion / Awesome-Diffusion-Models
  A collection of resources and papers on Diffusion Models
  ☆10,758Updated last month
• hojonathanho / diffusion
  Denoising Diffusion Probabilistic Models
  ☆3,637Updated last year
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,206Updated this week
• google-research / scenic
  Scenic: A Jax Library for Computer Vision Research and Beyond
  ☆3,248Updated 3 weeks ago
• YangLing0818 / Diffusion-Models-Papers-Survey-Taxonomy
  Diffusion model papers, survey, and taxonomy
  ☆2,894Updated last month
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,149Updated 2 weeks ago
• openai / guided-diffusion
  ☆6,068Updated 2 months ago
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,621Updated last year
• cmhungsteve / Awesome-Transformer-Attention
  An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
  ☆4,541Updated last month
• mlfoundations / open_clip
  An open source implementation of CLIP.
  ☆9,782Updated last month
• google-research / big_vision
  Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
  ☆2,206Updated 3 weeks ago
• adapter-hub / adapters
  A Unified Library for Parameter-Efficient and Modular Transfer Learning
  ☆2,525Updated 3 weeks ago
• openai / improved-diffusion
  Release for Improved Denoising Diffusion Probabilistic Models
  ☆3,136Updated 2 months ago
• facebookresearch / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆8,351Updated this week
• pytorch / captum
  Model interpretability and understanding for PyTorch
  ☆4,810Updated this week
• huggingface / trl
  Train transformer language models with reinforcement learning.
  ☆9,288Updated this week
• bitsandbytes-foundation / bitsandbytes
  Accessible large language models via k-bit quantization for PyTorch.
  ☆6,029Updated this week
• ashleve / lightning-hydra-template
  PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
  ☆4,092Updated last month
• google-research / vision_transformer
  ☆10,072Updated 3 months ago
• facebookresearch / multimodal
  TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.
  ☆1,423Updated this week
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆5,950Updated this week
• Lightning-AI / torchmetrics
  Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,077Updated this week