tummfm / jax-sgmcView external linksLinks
Stochastic Gradient MCMC for Jax
☆18May 19, 2025Updated 8 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☆36Feb 29, 2024Updated last year
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 7 years ago
- Training Neural Network potentials through customizable routines in JAX.☆61Jul 30, 2025Updated 6 months ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 3 years ago
- Action rules☆10Apr 22, 2025Updated 9 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Dec 9, 2025Updated 2 months ago
- Continual Resilient (CoRe) Optimizer for PyTorch☆11Jun 10, 2024Updated last year
- FORward MOdeling tool for Spectral Analysis☆10Dec 19, 2025Updated last month
- 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☆12Oct 20, 2025Updated 3 months ago
- Suite for Analysis of Molecular Simulations☆11Aug 28, 2025Updated 5 months ago
- Python implementation of Parker wind models for exoplanet atmospheres☆13May 12, 2025Updated 9 months ago
- can Polars crunch 27GBs of data faster than Pyspark?☆13Nov 6, 2023Updated 2 years ago
- A Python package for working with pipeline-produced spectra from IGRINS, HPF, and Keck NIRSPEC☆19Jan 12, 2026Updated last month
- Neural multiclass ab initio reconstruction for cryo-EM.☆13Dec 5, 2024Updated last year
- 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
- ☆23Oct 31, 2025Updated 3 months ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 4 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
- ☆11Oct 15, 2023Updated 2 years ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆16Dec 10, 2023Updated 2 years ago
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆35Nov 20, 2025Updated 2 months ago
- Repository for various academic and professional CV versions☆10Jan 16, 2026Updated 3 weeks ago
- JAX code for modeling nearly-Keplerian orbits☆17Jan 8, 2026Updated last month
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- ☆16Feb 17, 2025Updated 11 months ago
- ☆12Feb 7, 2026Updated last week
- A step-by-step tutorial to code the McMurchie-Davidson scheme to compute molecular integrals☆10Jul 4, 2021Updated 4 years ago
- Forward kinematics, inverse kinematics, Newton-Euler algorithm and derivations☆11Jan 12, 2017Updated 9 years ago
- ☆11Jul 25, 2021Updated 4 years ago
- Codes for translating structural defects to atomic properties☆14Sep 12, 2022Updated 3 years ago
- Benchopt benchmark for ResNet fitting on a classification task☆12Sep 19, 2023Updated 2 years ago
- Modern Python & C++ API for the radiative transfer solver DISORT. Parallelized with PyTorch tensors; compile-free with: pip install pydis…☆15Updated this week
- Modern, minimal, and modular LaTeX CV template ✨ 📄☆23Dec 4, 2025Updated 2 months ago
- A multidimensional spline interpolation by Google/JAX☆19Feb 13, 2024Updated 2 years ago