mancusolab / traceax
Stochastic trace estimation using JAX
☆11Updated 2 months ago
Related projects: ⓘ
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- ☆56Updated 2 years ago
- ☆28Updated this week
- ☆40Updated 2 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆39Updated 11 months ago
- Einsum-like high-level array sharding API for JAX☆31Updated 2 months ago
- ☆49Updated last year
- Graph Learning with JAX☆14Updated 2 years ago
- Exponential families for JAX☆54Updated last week
- Automated SMC with Probabilistic Program Proposals, for the Gen PPL.☆19Updated 10 months ago
- In which I learn about score functions and how they can be used to generate data.☆14Updated 5 months ago
- Inference Combinators in JAX☆42Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Tools for JAX☆44Updated this week
- A Simple Statistical Distribution Library in JAX☆16Updated 5 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated last year
- A collection of graph neural networks implementations in JAX☆30Updated 9 months ago
- Composable kernels for scikit-learn implemented in JAX.☆41Updated 3 years ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated this week
- Exact OU processes with JAX☆30Updated last month
- A metrics library for the JAX ecosystem☆36Updated last year
- ☆17Updated 4 months ago
- Stencil computations in JAX☆66Updated 11 months ago
- ☆18Updated 5 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆20Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year