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.☆73Updated this week
- Lux layers and utilities to build and train hybrid dynamic models.☆18Dec 5, 2025Updated 9 months ago
- Action rules☆12Aug 13, 2026Updated last month
- 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.
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 8 years ago
- ☆15Feb 17, 2025Updated last year
- 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.☆23Sep 10, 2026Updated 2 weeks ago
- can Polars crunch 27GBs of data faster than Pyspark?☆13Nov 6, 2023Updated 2 years ago
- Deep Coarse-grained Potentials via Relative Entropy Minimization☆18Feb 22, 2023Updated 3 years ago
- A comprehensive collection of 35+ recurrent neural network layers for Flux.jl☆29Sep 18, 2026Updated last week
- A software package for flexible HPC GPs☆16Aug 11, 2026Updated last month
- ☆22May 7, 2025Updated last year
- 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.
- ☆52Jun 8, 2026Updated 3 months ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 8 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆28Sep 15, 2026Updated last week
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆62Updated this week
- Coordinate Axes for JAX☆67Sep 9, 2026Updated 2 weeks ago
- Make Rmarkdown and Shiny apps fullscreen☆10Sep 1, 2021Updated 5 years ago
- MCHMC: sampler from an arbitrary differentiable distribution☆79Dec 9, 2025Updated 9 months ago
- This repository contains supplementary information and code to help users access the UniFoil dataset.☆15Apr 7, 2026Updated 5 months ago
- ☆22Sep 30, 2025Updated 11 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.
- ☆20Aug 18, 2026Updated last month
- Translate Latex Formulas to R Code☆14Sep 19, 2021Updated 5 years ago
- Bluesky optimization package☆24Updated this week
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Jun 22, 2026Updated 3 months ago
- Python implementation of Parker wind models for exoplanet atmospheres☆16Updated this week
- Evolutionary Individual based modelling, mathematically grounded.☆13Mar 25, 2022Updated 4 years ago
- Interpolation and function approximation with JAX☆282Updated this week
- An interior point solver for mixed complementarity problems, supporting CPU/GPU batched-parallel execution.☆20Aug 4, 2026Updated last month
- Rewriting the Stan compiler in OCaml☆23Jul 20, 2021Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 2 years ago
- Continual Resilient (CoRe) Optimizer for PyTorch☆12Jun 10, 2024Updated 2 years ago
- 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
- Suite for Analysis of Molecular Simulations☆12Sep 17, 2026Updated last week
- An interface for using NVIDIA's AMGX library from the Julia language.☆19Sep 11, 2026Updated 2 weeks ago
- PyTorch/PyG implementation of M3GNet☆12Sep 11, 2023Updated 3 years ago