daigo0927 / jax-ddimLinks
Jax/Flax implementation of Denoising Diffusion Implicit Models
☆20Updated 3 years ago
Alternatives and similar repositories for jax-ddim
Users that are interested in jax-ddim are comparing it to the libraries listed below
Sorting:
- minimal diffusion model for self-study☆28Updated 2 years ago
- Notionに毎日新しいarXiv論文のアブストラクト日本語訳 + αを表示するスクリプト☆12Updated 2 years ago
- Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)☆193Updated 3 years ago
- AIST Toolkit for Accelerating Machine Learning Research☆34Updated this week
- PyTorch implementations of normalizing flow and its variants.☆79Updated 4 years ago
- ☆11Updated 2 years ago
- CyberAgent AI Lab研修: "モデルコードの高速化・最適化チュートリアル"☆35Updated 9 months ago
- PyTorch implementation of slicing adversarial network (SAN)☆98Updated last month
- A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony☆33Updated 2 years ago
- Deep reinforcement learning library built on top of Neural Network Libraries☆127Updated last year
- Mamba training library developed by kotoba technologies☆69Updated last year
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆93Updated 2 years ago
- Official implementation of "TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models"☆120Updated 3 months ago
- ☆10Updated last year
- A differentiable argmax function for PyTorch☆42Updated 5 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆57Updated last year
- Flow Matching implemented in PyTorch☆85Updated last month
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆70Updated last month
- ☆49Updated last year
- A real-time and light-weight software for generation of non-linguistic behaviors (turn-taking, backchannel, and head-nodding) in conversa…☆74Updated last week
- The 2D discrete wavelet transform for JAX☆44Updated 2 years ago
- ☆44Updated last year
- ☆13Updated 2 years ago
- Simple transformer implementations that I can understand☆20Updated 4 years ago
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆63Updated last year
- FID computation in Jax/Flax.☆29Updated last year
- [Neurips 2021]Diffusion Normalizing Flow (DiffFlow)☆119Updated 2 years ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆101Updated last year
- xvector model on jtubespeech☆46Updated 2 years ago