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:
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 4 years ago
- 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.☆73Oct 7, 2026Updated last week
- Lux layers and utilities to build and train hybrid dynamic models.☆18Dec 5, 2025Updated 10 months ago
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 8 years 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.
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 4 years ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆25Oct 1, 2026Updated 2 weeks ago
- A software package for flexible HPC GPs☆16Updated this week
- ☆22May 7, 2025Updated last year
- ☆53Jun 8, 2026Updated 4 months ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 9 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆28Sep 23, 2026Updated 3 weeks ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆63Sep 26, 2026Updated 2 weeks ago
- LLM Benchmark problems for SWE tasks in julia☆16Nov 24, 2025Updated 10 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Coordinate Axes for JAX☆67Updated this week
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 years ago
- MCHMC: sampler from an arbitrary differentiable distribution☆79Dec 9, 2025Updated 10 months ago
- Coordinates in JAX☆38Oct 8, 2026Updated last week
- ☆16Sep 29, 2026Updated 2 weeks ago
- This repository contains supplementary information and code to help users access the UniFoil dataset.☆15Apr 7, 2026Updated 6 months ago
- ☆20Aug 18, 2026Updated last month
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 5 years ago
- Bluesky optimization package☆25Sep 30, 2026Updated 2 weeks ago
- 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.
- Python implementation of Parker wind models for exoplanet atmospheres☆16Sep 21, 2026Updated 3 weeks ago
- Evolutionary Individual based modelling, mathematically grounded.☆13Mar 25, 2022Updated 4 years ago
- Interpolation and function approximation with JAX☆281Updated this week
- An interior point solver for mixed complementarity problems, supporting CPU/GPU batched-parallel execution.☆20Aug 4, 2026Updated 2 months ago
- Rewriting the Stan compiler in OCaml☆23Jul 20, 2021Updated 5 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 2 years ago
- This repository contains Matlab code for conducting geostatistical inverse modeling using large atmospheric datasets.☆10Aug 28, 2025Updated last year
- A step-by-step tutorial to code the McMurchie-Davidson scheme to compute molecular integrals☆10Jul 4, 2021Updated 5 years ago
- Inverse modelling framework for dynamical systems characterised by complex dynamics.☆13Oct 16, 2025Updated 11 months ago
- 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.
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated last year
- Julia's HPC command for automatic feature/model selection using all-subset-regression approaches☆18Sep 29, 2026Updated 2 weeks ago
- Suite for Analysis of Molecular Simulations☆12Sep 17, 2026Updated 3 weeks ago
- PyTorch/PyG implementation of M3GNet☆12Sep 11, 2023Updated 3 years ago
- Program Package for Sampling, Training and Applying ML-based Potential models☆13Oct 5, 2026Updated last week
- In this paper, we propose Filter Gradient Decent (FGD), an efficient stochastic optimization algorithm that makes a consistent estimation…☆12May 18, 2021Updated 5 years ago
- ☆67Jul 8, 2026Updated 3 months ago