Stochastic Gradient MCMC for Jax
☆20May 19, 2025Updated last year
Alternatives and similar repositories for jax-sgmc
Users that are interested in jax-sgmc are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆38Feb 29, 2024Updated 2 years ago
- Training Neural Network potentials through customizable routines in JAX.☆74Oct 23, 2026Updated 2 weeks ago
- chemtrain-deploy: A parallel and scalable framework for machine learning potentials in million-atom MD simulations☆11Jan 15, 2026Updated 9 months ago
- ☆15Feb 17, 2025Updated last year
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A comprehensive collection of 35+ recurrent neural network layers for Flux.jl☆29Sep 18, 2026Updated last month
- Deep Coarse-grained Potentials via Relative Entropy Minimization☆18Feb 22, 2023Updated 3 years ago
- A software package for flexible HPC GPs☆16Updated this week
- ☆22May 7, 2025Updated last year
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆25Updated this week
- ☆53Jun 8, 2026Updated 4 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆28Sep 23, 2026Updated last month
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆63Updated this week
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 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.
- Coordinate Axes for JAX☆67Oct 12, 2026Updated 3 weeks ago
- MCHMC: sampler from an arbitrary differentiable distribution☆79Dec 9, 2025Updated 10 months ago
- Coordinates in JAX☆38Updated this week
- Generates (almost) evenly distributed, equidistant points across a perfect sphere or the globe☆10Jun 30, 2018Updated 8 years ago
- Fully and Partially Bayesian Neural Nets☆84Apr 30, 2025Updated last year
- ☆16Updated this week
- ☆12May 8, 2025Updated last year
- microJAX: A GPU-accelerated differentiable microlensing modeling framework☆14Oct 29, 2026Updated last week
- An interior point solver for mixed complementarity problems, supporting CPU/GPU batched-parallel execution.☆20Aug 4, 2026Updated 3 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆587Oct 27, 2026Updated last week
- ☆20Aug 18, 2026Updated 2 months ago
- Algebraic multigrid solvers in JAX☆43Aug 27, 2026Updated 2 months ago
- Persistent Sampling for fast Bayesian posterior sampling and model evidence estimation in scientific applications☆14Jun 5, 2026Updated 5 months ago
- An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)☆10Dec 19, 2024Updated last year
- AI agent config compatibility matrix — which configuration files are read by which AI coding tools (April 2026)☆14Aug 26, 2026Updated 2 months ago
- Dynamic Neural Cellular Automata. Used primarily in my PhD research. Efficiently learn cellular automata rules that give desired emergent…☆15May 24, 2024Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 3 years ago
- Example astronomy data analysis problems tackled using probabilistic models☆13Aug 25, 2021Updated 5 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A Python interface for implementation of crystal structure pre-relaxation and prediction using machine-learning inter-atomic potentials.☆11May 20, 2024Updated 2 years ago
- Interpolation and function approximation with JAX☆283Updated this week
- SPAGeDi: Spatial Pattern Analysis of Genetic Diversity☆10Nov 27, 2018Updated 7 years ago
- A project to illustrate the power of Jax for simulating astrochemical kinetics.☆11Oct 12, 2026Updated 3 weeks ago
- Bayesian Deep Ensembles via MILE: easy to use, scikit-learn compatible and fast (JAX powered)☆46Aug 12, 2026Updated 2 months ago
- ☆11Mar 9, 2022Updated 4 years ago
- Continual Resilient (CoRe) Optimizer for PyTorch☆12Jun 10, 2024Updated 2 years ago