Alternatives and similar repositories for jax-ddim

Users that are interested in jax-ddim are comparing it to the libraries listed below

• Kei18 / tiny-tiny-diffusion
  minimal diffusion model for self-study
  ☆28Updated 2 years ago
• sony / sqvae
  Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)
  ☆193Updated 3 years ago
• aistairc / aiaccel
  AIST Toolkit for Accelerating Machine Learning Research
  ☆34Updated this week
• tatsy / normalizing-flows-pytorch
  PyTorch implementations of normalizing flow and its variants.
  ☆79Updated 4 years ago
• discus0434 / notion-news
  Notionに毎日新しいarXiv論文のアブストラクト日本語訳 + αを表示するスクリプト
  ☆12Updated 2 years ago
• CyberAgentAILab / model-acceleration-tutorial
  CyberAgent AI Lab研修： "モデルコードの高速化・最適化チュートリアル"
  ☆35Updated 9 months ago
• tennmoku71 / advent_calendar_cyberagent_llm_dialogue_system
  ☆11Updated last year
• david-wb / softargmax
  A differentiable argmax function for PyTorch
  ☆42Updated 5 years ago
• kotoba-tech / kotomamba
  Mamba training library developed by kotoba technologies
  ☆70Updated last year
• YutaroOgawa / ddpm_cifer10
  拡散モデルを学びたい初学者向けです。書籍「コンピュータビジョン最前線 Summer 2023」の「イマドキノ拡散モデル」の解説をベースに、CIFER-10で画像生成をします
  ☆20Updated 2 years ago
• SakanaAI / TAID
  Official implementation of "TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models"
  ☆119Updated 2 months ago
• koshian2 / ai-art-book
  ☆13Updated 2 years ago
• sony / san
  PyTorch implementation of slicing adversarial network (SAN)
  ☆98Updated 2 weeks ago
• yukara-ikemiya / minimal-sqvae
  A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony
  ☆33Updated 2 years ago
• iwiwi / epochraft
  Checkpointable dataset utilities for foundation model training
  ☆32Updated last year
• apple / ml-agm
  ☆44Updated last year
• SakanaAI / CycleQD
  CycleQD is a framework for parameter space model merging.
  ☆45Updated 10 months ago
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆59Updated 2 years ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆68Updated 3 weeks ago
• keishihara / flow-matching
  Flow Matching implemented in PyTorch
  ☆81Updated 2 weeks ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• kvfrans / splus
  ☆121Updated 6 months ago
• alfredplpl / clean-diffusion
  Latent Diffusion Model made of public domain images (CC-0).
  ☆35Updated 3 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆82Updated 3 years ago
• lucidrains / perceiver-ar-pytorch
  Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch
  ☆93Updated 2 years ago
• luchris429 / DiscoPOP
  Code for Discovering Preference Optimization Algorithms with and for Large Language Models
  ☆64Updated last year
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆29Updated last year
• qsh-zh / DiffFlow
  [Neurips 2021]Diffusion Normalizing Flow (DiffFlow)
  ☆119Updated 2 years ago
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆135Updated 2 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆120Updated 3 months ago