Hackable pure-torch implementation of diffusion models
☆23Sep 21, 2024Updated last year
Alternatives and similar repositories for tiny-diff
Users that are interested in tiny-diff are comparing it to the libraries listed below
Sorting:
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆31Mar 21, 2024Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆37Mar 18, 2025Updated 11 months ago
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated 11 months ago
- blender scripts for shapenet☆11Oct 12, 2020Updated 5 years ago
- A Pytorch Lightning WGAN-gp to generate faces☆11Jan 26, 2021Updated 5 years ago
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- Baseline solution for ADC2023 ECML competition☆11Apr 14, 2023Updated 2 years ago
- ☆10Sep 21, 2024Updated last year
- A grab-bag of utilities for messing around with filesystems and 3D models☆13Nov 18, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated 9 months ago
- Tutorial kit for building a 3D deep reinforcement learning environment with Unity ML-Agents.☆11Oct 22, 2021Updated 4 years ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 3 years ago
- Code base for NeurIPS 2022 paper Curriculum Reinforcement Learning using Optimal Transport via Gradual Domain Adaptation.☆11Aug 21, 2023Updated 2 years ago
- RL Environment and Benchmark pipeline. Code accompanying paper "Neurophysiologically Realistic Environment for Comparing Adaptive Deep Br…☆19Feb 6, 2026Updated 3 weeks ago
- ☆18Mar 21, 2025Updated 11 months ago
- Codebase for Extracting Reward Functions from Diffusion Models☆16Dec 7, 2023Updated 2 years ago
- This is a repo covers ai research papers pseudocodes☆17Jun 20, 2023Updated 2 years ago
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆63Dec 19, 2025Updated 2 months ago
- ☆18Nov 8, 2023Updated 2 years ago
- [IEEE Transactions on Intelligent Transportation Systems] Curricular Subgoal for Inverse Reinforcement Learning☆16Jul 31, 2023Updated 2 years ago
- IMFine: 3D Inpainting via Geometry-guided Multi-view Refinement☆18Mar 7, 2025Updated 11 months ago
- Generative Modeling via Drifting in MLX☆42Feb 6, 2026Updated 3 weeks ago
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆20Dec 1, 2025Updated 3 months ago
- The code for the video tutorial series on building a Transformer from scratch: https://www.youtube.com/watch?v=XR4VDnJzB8o☆19Apr 15, 2023Updated 2 years ago
- Reading list for research topics in Diffusion models.☆18Jan 12, 2024Updated 2 years ago
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆21Jun 22, 2023Updated 2 years ago
- An implementation of AlphaZero and MCTS with neural networks for Tetris☆22Mar 21, 2025Updated 11 months ago
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- Pytorch implementation of DGflow (ICLR 2021).☆16Mar 3, 2021Updated 4 years ago
- Extracting high-resolution DINOv2 features for data-scarce applications like materials segmentation.☆27Nov 10, 2025Updated 3 months ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Jul 21, 2023Updated 2 years ago
- CENG502 Spring 2022 Projects☆20May 5, 2023Updated 2 years ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆25Dec 5, 2023Updated 2 years ago
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- ☆29Apr 10, 2025Updated 10 months ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆40Jul 13, 2025Updated 7 months ago
- Deep Generative Models course, 2021☆22Dec 25, 2021Updated 4 years ago