Discontinuous Hamiltonian Monte Carlo in JAX
☆42Feb 24, 2020Updated 6 years ago
Alternatives and similar repositories for dhmc-jax
Users that are interested in dhmc-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 6 years ago
- "Variational inference tools to leverage estimator sensitivity."☆15May 11, 2023Updated 3 years ago
- Python bindings to some optimization benchmarks (robotics problems), in order to constrained optimization solvers. Includes also an inter…☆21Dec 8, 2022Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 6 years ago
- Code for Semi-crowdsourced Clustering with Deep Generative Models☆12Dec 9, 2022Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Samplers from the paper "Stochastic Gradient MCMC with Repulsive Forces"☆11Feb 3, 2020Updated 6 years ago
- Julia implementation of LULESH with MPI + X.☆12Jul 23, 2022Updated 4 years ago
- Pioupiou 🐤, the nano PPL☆11May 25, 2022Updated 4 years ago
- Featherweight embedded probabilistic programming language and compositional inference programming library☆19May 15, 2020Updated 6 years ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Oct 5, 2021Updated 5 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- Exponential families for JAX☆77Updated this week
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆85Jul 15, 2020Updated 6 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Differentiable N-body code (JAX)☆11May 8, 2025Updated last year
- A differentiable astrophysics simulator in JAX☆12Oct 27, 2026Updated last week
- Apply the Nested Sampling Monte Carlo algorithm to fit exoplanet radial velocity data and estimate the posterior distribution of the mod…☆11May 8, 2025Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated last year
- ☆13Aug 17, 2020Updated 6 years ago
- ☆32Jan 24, 2018Updated 8 years ago
- Monte Carlo thermal radiation transport code☆12Sep 23, 2026Updated last month
- Variational Inference by Policy Search☆13Apr 24, 2019Updated 7 years ago
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years 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.
- ☆16Jun 27, 2021Updated 5 years ago
- Codes associated with Warne, Baker and Simpson (2018) Implementations of simulation and inference algorithms for stochastic reaction netw…☆10Apr 13, 2021Updated 5 years ago
- Code to reproduce the results of 👇☆18Aug 19, 2022Updated 4 years ago
- ☆242Sep 3, 2023Updated 3 years ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆10Mar 22, 2021Updated 5 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 3 years ago
- Read, write and manipulate code which reads, writes and manipulates code.☆11Mar 15, 2020Updated 6 years ago
- Monotone operator equilibrium networks☆54Jun 22, 2020Updated 6 years ago
- ☆18Mar 20, 2022Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Artificial Neural Network for Image Classification☆13May 8, 2025Updated last year
- Compiler with automatic differentiation☆50Oct 18, 2023Updated 3 years ago
- Bilinear interpolation on grids with jax☆16May 13, 2022Updated 4 years ago
- ☆154May 25, 2020Updated 6 years ago
- ☆18Apr 22, 2024Updated 2 years ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 5 years ago