Alternatives and similar repositories for ml-tract

Users that are interested in ml-tract are comparing it to the libraries listed below

• apple / ml-duet
  DUET: 2D Structured and Approximately Equivariant Representations, ICML 2023
  ☆18Updated last year
• apple / ml-agm
  ☆44Updated last year
• apple / ml-spin
  This repository contains the official implementation for the ECCV'22 paper, "SPIN: An Empirical Evaluation on Sharing Parameters of Isotr…
  ☆20Updated last year
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated last year
• apple / ml-jointnet
  ☆27Updated last year
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆114Updated 8 months ago
• lucidrains / recurrent-interface-network-pytorch
  Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…
  ☆204Updated last year
• apple / ml-dab
  ☆23Updated 3 years ago
• AndyShih12 / paradigms
  PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight
  ☆142Updated last year
• mlfoundations / open-diffusion
  Simple large-scale training of stable diffusion with multi-node support.
  ☆133Updated 2 years ago
• FutureXiang / edm2
  Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆32Updated last year
• philippe-eecs / vitok
  ☆32Updated last month
• apple / ml-ademamix
  ☆61Updated 7 months ago
• lukemelas / fixed-point-diffusion-models
  ☆54Updated last year
• openai / consistency_models_cifar10
  Consistency models trained on CIFAR-10, in JAX.
  ☆147Updated last year
• CompVis / retrieval-augmented-diffusion-models
  Official codebase for the Paper “Retrieval-Augmented Diffusion Models”
  ☆131Updated 2 years ago
• apple / ml-collegial-ensembles
  ☆11Updated 3 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆83Updated 2 years ago
• apple / ml-selfcond
  Self-Conditioning Pre-Trained Language Models, ICML 2022
  ☆31Updated 2 years ago
• baaivision / MUSE-Pytorch
  An in-context conditioning version of MUSE with pre-trained checkpoints.
  ☆112Updated 2 years ago
• apple / ml-entropy-reconstruction
  ☆29Updated last year
• dongzhuoyao / uspace
  An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"
  ☆42Updated last year
• apple / ml-pgdvs
  [ICLR 2024] Official implementation of "Pseudo-Generalized Dynamic View Synthesis from a Video"
  ☆30Updated 8 months ago
• apple / ml-ogen
  ☆13Updated last year
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆105Updated 3 weeks ago
• apple / ml-diffusion-alignment-gflownet
  PyTorch implementation for our paper "Improving GFlowNets for Text-to-Image Diffusion Alignment."
  ☆27Updated 9 months ago
• pcuenca / lpips-j
  Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
  ☆15Updated 2 years ago
• apple / ml-dr
  A light-weight implementation of ICCV2023 paper "Reinforce Data, Multiply Impact: Improved Model Accuracy and Robustness with Dataset Rei…
  ☆79Updated last year