Diffusion Probabilistic Model in Jax
☆13Apr 20, 2024Updated last year
Alternatives and similar repositories for diffusion-jax
Users that are interested in diffusion-jax are comparing it to the libraries listed below
Sorting:
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Jan 28, 2024Updated 2 years ago
- Official Implementation of Energy Transformer in PyTorch for Mask Image Reconstruction☆32Mar 10, 2024Updated 2 years ago
- Supposed you've downloaded CelebA & CelebA-HQ dataset, and want to get HQ images from them.☆22Aug 21, 2019Updated 6 years ago
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆41Mar 7, 2026Updated 2 weeks ago
- ☆21Apr 2, 2024Updated last year
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆40Jan 27, 2026Updated last month
- Emacs AI-powered text & code completion client☆23Jun 16, 2025Updated 9 months ago
- Curated list of JAX Resources and Packages☆35Mar 2, 2026Updated 2 weeks ago
- ☆15Sep 4, 2025Updated 6 months ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- Convert between robotics dataset formats (RLDS, LeRobot v2/v3, Zarr, HDF5, Rosbag). Inspect, visualize, and analyze datasets. Works with …☆73Mar 11, 2026Updated last week
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆50Aug 6, 2024Updated last year
- ☆28Feb 23, 2026Updated 3 weeks ago
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆31Mar 6, 2026Updated 2 weeks ago
- Investigating and Defending Shortcut Learning in Personalized Diffusion Models☆13Nov 19, 2024Updated last year
- This is a python library module with code to help make working with Soar SML in python easier.☆21Jul 24, 2023Updated 2 years ago
- USD models for NVIDIA Isaac Sim☆16Aug 8, 2025Updated 7 months ago
- PyTorch implementation of CtRNet https://arxiv.org/abs/2302.14332☆29Sep 9, 2024Updated last year
- Soar Agent (and SML code) that learns through situated interactive instruction in a robotic environment☆35Sep 15, 2023Updated 2 years ago
- Short-time Fourier transform (STFT) for JAX☆15Dec 20, 2021Updated 4 years ago
- Feed-forward compressor experiments source code for "Differentiable All-pole Filters for Time-varying Audio Systems".☆22Jun 10, 2024Updated last year
- Does Diffusion Beat GAN in Image Super Resolution?☆12May 27, 2024Updated last year
- Sample audio and video files for the YouTube Video Tutorials on HTML5 Audio and Video☆16Mar 4, 2021Updated 5 years ago
- [ECCV2024] Immunizing text-to-image Models against Malicious Adaptation☆17Jan 17, 2025Updated last year
- Sequential Monte Carlo sampler for PyMC2 models.☆13Apr 4, 2018Updated 7 years ago
- ☆16Sep 19, 2023Updated 2 years ago
- A curated list of awesome Unlearnable Example papers resources.☆13Dec 14, 2025Updated 3 months ago
- ☆11Jun 17, 2016Updated 9 years ago
- Probabilistic inference for models of behaviour☆10Mar 5, 2026Updated 2 weeks ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- ☆28Jan 8, 2026Updated 2 months ago
- A generic sampling-based MPC python library based on jax☆37Jan 30, 2026Updated last month
- Possibly the most obscure piece of Nix magic yet☆13Jan 30, 2023Updated 3 years ago
- ☆15Jul 21, 2023Updated 2 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- Open source Java framework to create, process and manage mixtures of exponential family☆14Aug 4, 2015Updated 10 years ago
- Async behavior tree for python☆42Jan 7, 2026Updated 2 months ago
- Official TensorFlow-based implementation of adversarial attack for super-resolution models☆18Mar 22, 2021Updated 4 years ago
- One-shot Generative Prior in Hankel-k-space for Parallel Imaging Reconstruction☆12Dec 4, 2024Updated last year