Stochastic Gradient MCMC for Jax
☆18May 19, 2025Updated 9 months ago
Alternatives and similar repositories for jax-sgmc
Users that are interested in jax-sgmc are comparing it to the libraries listed below
Sorting:
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 3 years ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆37Feb 29, 2024Updated 2 years ago
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 7 years ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 3 years ago
- Training Neural Network potentials through customizable routines in JAX.☆63Feb 10, 2026Updated 3 weeks ago
- Action rules☆10Apr 22, 2025Updated 10 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Dec 9, 2025Updated 2 months ago
- Official code repository for the MICCAI 2025 paper "UltraRay: Introducing Full-Path Ray Tracing in Physics-Based Ultrasound Simulation"☆17Aug 13, 2025Updated 6 months ago
- Continual Resilient (CoRe) Optimizer for PyTorch☆11Jun 10, 2024Updated last year
- This repository contains Matlab code for conducting geostatistical inverse modeling using large atmospheric datasets.☆10Aug 28, 2025Updated 6 months ago
- Suite for Analysis of Molecular Simulations☆11Aug 28, 2025Updated 6 months ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- Program Package for Sampling, Training and Applying ML-based Potential models☆12Feb 25, 2026Updated last week
- FORward MOdeling tool for Spectral Analysis☆11Updated this week
- JAX code for modeling nearly-Keplerian orbits☆17Jan 8, 2026Updated last month
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- In this paper, we propose Filter Gradient Decent (FGD), an efficient stochastic optimization algorithm that makes a consistent estimation…☆12May 18, 2021Updated 4 years ago
- Python implementation of Parker wind models for exoplanet atmospheres☆13May 12, 2025Updated 9 months ago
- A Python package for working with pipeline-produced spectra from IGRINS, HPF, and Keck NIRSPEC☆19Jan 12, 2026Updated last month
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆11Oct 15, 2023Updated 2 years ago
- Data relevant to the article "Machine learning determination of atomic dynamics at grain boundaries" https://arxiv.org/abs/1803.01416☆11Oct 2, 2018Updated 7 years ago
- can Polars crunch 27GBs of data faster than Pyspark?☆13Nov 6, 2023Updated 2 years ago
- Neural multiclass ab initio reconstruction for cryo-EM.☆13Dec 5, 2024Updated last year
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 years ago
- Repository for various academic and professional CV versions☆10Jan 16, 2026Updated last month
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- ☆24Oct 31, 2025Updated 4 months ago
- Modern Python & C++ API for the radiative transfer solver DISORT. Parallelized with PyTorch tensors; compile-free with: pip install pydis…☆15Updated this week
- This is the official implementation for paper "On Powerful Ways to Generate: Autoregression, Diffusion, and Beyond".☆20Nov 17, 2025Updated 3 months ago
- PyTorch/PyG implementation of M3GNet☆11Sep 11, 2023Updated 2 years ago
- ☆12Feb 15, 2026Updated 2 weeks ago
- ☆11Jul 25, 2021Updated 4 years ago
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 2 years ago
- Codes for translating structural defects to atomic properties☆14Sep 12, 2022Updated 3 years ago
- Forward kinematics, inverse kinematics, Newton-Euler algorithm and derivations☆11Jan 12, 2017Updated 9 years ago
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆38Feb 18, 2026Updated 2 weeks ago
- ☆16Feb 17, 2025Updated last year