flowersteam / sbmltoodejax
☆11Updated 4 months ago
Alternatives and similar repositories for sbmltoodejax
Users that are interested in sbmltoodejax are comparing it to the libraries listed below
Sorting:
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Exact OU processes with JAX☆43Updated last month
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Learn one, get them all for free☆12Updated last year
- An implementation of ESM2 in Equinox+JAX☆25Updated 3 months ago
- ☆36Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Matrix-free linear algebra in JAX.☆118Updated 4 months ago
- Implementation of various equivariant models in JAX☆12Updated last year
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆15Updated last year
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated last year
- ☆30Updated 8 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- ☆27Updated last year
- A collection of graph neural networks implementations in JAX☆32Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Riemannian Optimization Using JAX☆49Updated last year
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 3 weeks ago
- A software package for flexible HPC GPs☆16Updated this week
- Stencil computations in JAX☆71Updated last year
- ☆50Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆52Updated 2 weeks ago
- A Wadler--Lindig pretty printer for Python☆39Updated this week
- Multiple dispatch over abstract array types in JAX.☆121Updated 2 weeks ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 11 months ago
- A generic interface for linear algebra backends☆73Updated 2 months ago
- Implementation of Forward Laplacian algorithm in JAX☆66Updated 2 weeks ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆46Updated last month