homerjed / sbgmView external linksLinks
Score-based Diffusion models in JAX.
☆17Dec 29, 2025Updated last month
Alternatives and similar repositories for sbgm
Users that are interested in sbgm are comparing it to the libraries listed below
Sorting:
- Implementation of Computer Vision Models in JAX (equinox)☆18Jan 15, 2026Updated last month
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 2 years ago
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 5 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Jul 15, 2025Updated 7 months ago
- ☆27Jun 3, 2024Updated last year
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 6 months ago
- SC24 Deep Learning at Scale Tutorial Material☆34Feb 5, 2025Updated last year
- [NeurIPS 2025] Geometry Aware Operator Transformer As An Efficient And Accurate Neural Surrogate For PDEs On Arbitrary Domains☆73Oct 23, 2025Updated 3 months ago
- They do indeed go brrrr☆30May 18, 2025Updated 8 months ago
- Action rules☆10Apr 22, 2025Updated 9 months ago
- Repository of notes, code and notebooks in Python for the book "Reinforcement Learning: An Introduction" by Richard S. Sutton and Andrew …☆36Aug 23, 2025Updated 5 months ago
- Simulation framework for a pixelated Liquid Argon TPC☆10Feb 3, 2026Updated last week
- Kinematic and dynamic models of continuum and articulated soft robots.☆15Nov 22, 2025Updated 2 months ago
- Python calculator for the Wigner D matrices, Clebsch-Gordan coefficients, hydrogen atoms, and irreducible representations of the Lorentz …☆11Mar 22, 2023Updated 2 years ago
- Plug-and-Play Benchmarking of Reinforcement Learning Algorithms for Large-Scale Flow Control☆29Feb 4, 2026Updated last week
- Scaling safe exploration to vision control☆14Feb 19, 2025Updated 11 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated 3 weeks ago
- ☆14Aug 20, 2024Updated last year
- Virtual Bash interpreter with a virtual file system for multi-tenant environments.☆31Updated this week
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Jan 19, 2025Updated last year
- Hands-on tutorial about Meta RL and GP-MPC at the RL4AA'24 workshop.☆15May 10, 2024Updated last year
- ☆13Oct 13, 2024Updated last year
- Fast semantic search for biorXiv manuscripts☆12Feb 16, 2025Updated 11 months ago
- ☆20Jul 23, 2025Updated 6 months ago
- Advanced futures library☆16Feb 3, 2026Updated last week
- Provide retain_mut method that has the same functionality as retain but gives mutable borrow to the predicate☆13May 20, 2022Updated 3 years ago
- ☆13Nov 18, 2023Updated 2 years ago
- can Polars crunch 27GBs of data faster than Pyspark?☆13Nov 6, 2023Updated 2 years ago
- ☆11Jul 2, 2024Updated last year
- The public repo for the source code of WindBorne's WeatherMesh-3☆20Oct 16, 2025Updated 3 months ago
- A fast code for long-baseline neutrino oscillation probabilities in matter☆13Oct 27, 2025Updated 3 months ago
- A nascent Jax-based package for virtual brain modeling.☆12Feb 3, 2026Updated last week
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Dec 19, 2025Updated last month
- Samosa helps developers prioritize what needs to be tested.☆12Feb 23, 2023Updated 2 years ago
- A boilerplate for seamlessly integrating PyTorch's Distributed Data Parallel (DDP) with SLURM job scheduling and Weights and Biases. Kick…☆10Aug 21, 2023Updated 2 years ago
- Decoupled Neural Interfaces Using Synthetic Gradients - under develeopment☆11Jun 27, 2025Updated 7 months ago
- Dynamical Systems with JAX☆12Jan 11, 2026Updated last month
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 5 months ago
- Implementation of Diffusion Policy☆13Dec 13, 2024Updated last year