Diffusion Probabilistic Model in Jax
☆11Apr 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 last year
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆41May 6, 2025Updated 9 months ago
- ☆14Sep 29, 2025Updated 5 months ago
- Probabilistic inference for models of behaviour☆10Oct 13, 2025Updated 4 months ago
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆50Aug 6, 2024Updated last year
- 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
- Minimalist version of probml/rebayes☆10Sep 15, 2025Updated 5 months ago
- My experimental NixOS configuration for VisionFive 2☆14Feb 17, 2024Updated 2 years ago
- Investigating and Defending Shortcut Learning in Personalized Diffusion Models☆13Nov 19, 2024Updated last year
- An implementation of several unsupervised object discovery models (Slot Attention, SLATE, GNM) in PyTorch with pre-trained models.☆14May 26, 2025Updated 9 months ago
- Install MinionPro for Mac and TeX Live 2018☆13Oct 4, 2019Updated 6 years ago
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆27Updated this week
- Sequential Monte Carlo sampler for PyMC2 models.☆13Apr 4, 2018Updated 7 years ago
- ☆15Jan 3, 2024Updated 2 years ago
- ☆13Jul 9, 2024Updated last year
- General framework for Bayesian inversion of continuous hierarchical models☆10Sep 20, 2021Updated 4 years ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- StableFused is a toy library for experimenting with Diffusion Models, inspired by various sources.☆14Nov 2, 2023Updated 2 years ago
- minimalist vector ad☆11Feb 11, 2024Updated 2 years ago
- Does Diffusion Beat GAN in Image Super Resolution?☆12May 27, 2024Updated last year
- Possibly the most obscure piece of Nix magic yet☆13Jan 30, 2023Updated 3 years ago
- dcEmb, the Embecosm Dynamic Causal Modelling library☆13Aug 26, 2024Updated last year
- Convert between robotics dataset formats (RLDS, LeRobot v2/v3, Zarr, HDF5, Rosbag). Inspect, visualize, and analyze datasets. Works with …☆40Feb 14, 2026Updated 2 weeks ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 7 months ago
- Open-source MATLAB to Python Compiler (from 2008)☆15Jun 18, 2025Updated 8 months ago
- ☆11Jun 17, 2016Updated 9 years ago
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆35Jan 27, 2026Updated last month
- ☆15Jul 21, 2023Updated 2 years ago
- ☆16Sep 19, 2023Updated 2 years ago
- ☆21Oct 22, 2025Updated 4 months ago
- Code and data for Feher da Silva, C., Hare, T.A. Humans primarily use model-based inference in the two-stage task. Nat Hum Behav (2020). …☆13Jul 25, 2023Updated 2 years ago
- The first open-sourced personalized agent☆28Dec 1, 2025Updated 2 months ago
- Open source Java framework to create, process and manage mixtures of exponential family☆14Aug 4, 2015Updated 10 years ago
- [ECCV2024] Immunizing text-to-image Models against Malicious Adaptation☆17Jan 17, 2025Updated last year
- Made with LaTex. NENU's recommendation letter template.☆13May 26, 2024Updated last year
- NixOS configs☆11Oct 26, 2020Updated 5 years ago
- ☆16Jan 26, 2023Updated 3 years ago
- A library for visualizing and animating PDDL domains.☆15Sep 22, 2024Updated last year