Score-based Diffusion models in JAX.
☆19Dec 29, 2025Updated 10 months ago
Alternatives and similar repositories for sbgm
Users that are interested in sbgm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- diffusionjax is a simple and accessible diffusion models package in JAX☆52Jan 19, 2025Updated last year
- A python based framework for solving the shallow water equations.☆10Jan 4, 2021Updated 5 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆60Oct 8, 2026Updated 3 weeks ago
- SC24 Deep Learning at Scale Tutorial Material☆35Feb 5, 2025Updated last year
- Implementation of Deep Learning Models in JAX (equinox)☆25Oct 26, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆13Nov 18, 2023Updated 2 years ago
- Dataloader designed for JAX without any other dependencies☆15Feb 28, 2026Updated 8 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆23Jun 29, 2026Updated 4 months ago
- Code for the paper: Comparing and Contrasting Deep Learning Weather Prediction Backbones on Navier-Stokes and Atmospheric Dynamics☆17Aug 9, 2024Updated 2 years ago
- Parallel hyperparameter tuning with JAX☆40Sep 30, 2026Updated last month
- Experiment with diffusion models that you can run on your local jupyter instances☆67Oct 27, 2024Updated 2 years ago
- Large Eddy Simulation in JAX☆18Mar 14, 2026Updated 7 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- Official Implementation of "Variational Flow Maps: Make Some Noise for One-Step Conditional Generation"☆32Aug 7, 2026Updated 3 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 2 months ago
- They do indeed go brrrr☆30May 18, 2025Updated last year
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Score-based Diffusion for Super-Resolution☆12Jun 24, 2025Updated last year
- Official repo for Trumpets: Injective Flows for Inference and Inverse Problems☆15Jun 2, 2021Updated 5 years ago
- An FL algorithm inspired by FedGMA☆11Oct 21, 2023Updated 3 years ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆52Jun 26, 2025Updated last year
- A minimal poster template in Typst.☆28Oct 24, 2025Updated last year
- Implementation of Attention Schema-based Attention Control (ASAC), A Cognitive-Inspired Approach for Attention Management in Transformers☆18Jul 10, 2026Updated 3 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Neuroscience-inspired memory for AI agents — sleep cycles that rewrite and strengthen knowledge over time. TypeScript, PostgreSQL, MCP.☆14Oct 26, 2026Updated last week
- Knowledgebase— a collection of information for quantitative finance, insurance, mathematics and AI—This serves as a sprawling notebook of…☆18Updated this week
- ☆11Mar 31, 2023Updated 3 years ago
- Code accompanying paper "Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Sets."☆31Oct 11, 2022Updated 4 years ago
- Sample-Efficient Parametric Learning from Natural Language☆13Feb 5, 2026Updated 9 months ago
- Official Implementation of the paper "Variational Control for Guidance in Diffusion Models" accepted at ICML'25☆20Dec 5, 2025Updated 11 months ago
- A lightweight, parallel python wrapper for the GSL integration library☆15Mar 3, 2021Updated 5 years ago
- LearningByDoing NeurIPS 2021 Competition: Standalone Code and Results☆20Aug 30, 2022Updated 4 years ago
- A Variational Autoencoder in PyTorch for the CelebA Dataset.☆20Nov 1, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- NoiseGrad (and its extension NoiseGrad++) is a method to enhance explanations of artificial neural networks by adding noise to model weig…☆22May 11, 2023Updated 3 years ago
- Implementation of GenMimic, "From Generated Human Videos to Physically Plausible Robot Trajectories"☆19Sep 1, 2026Updated 2 months ago
- A library for coreset algorithms, written in JAX for fast execution and GPU support.☆43Oct 30, 2026Updated last week
- ☆11Jan 29, 2025Updated last year
- ☆26Feb 25, 2025Updated last year
- Fast, lightweight and parallelised simulation-based inference in JAX.☆24Oct 28, 2025Updated last year
- A fast code for long-baseline neutrino oscillation probabilities in matter☆15Jun 29, 2026Updated 4 months ago