dasayan05 / iclr24_blog_code
Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
☆15Updated last year
Alternatives and similar repositories for iclr24_blog_code:
Users that are interested in iclr24_blog_code are comparing it to the libraries listed below
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆41Updated 2 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 3 months ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆26Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- Generate bird's-eye views of conference proceedings.☆23Updated last month
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated 11 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 7 months ago
- ☆31Updated last month
- Learn online intrinsic rewards from LLM feedback☆33Updated last month
- ☆37Updated 2 years ago
- Repo to reproduce the First-Explore paper results☆37Updated 3 weeks ago
- Clean RL implementation using MLX☆28Updated 10 months ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 3 months ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- ☆26Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 6 months ago
- Deep Networks Grok All the Time and Here is Why☆23Updated 8 months ago
- FID computation in Jax/Flax.☆26Updated 6 months ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆17Updated 2 months ago
- ☆58Updated 2 years ago
- ☆13Updated 6 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆17Updated last year
- Jax like function transformation engine but micro, microjax☆30Updated 2 months ago
- ☆40Updated last month
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago