gchq / coreax
A library for coreset algorithms, written in Jax for fast execution and GPU support.
☆25Updated this week
Related projects ⓘ
Alternatives and complementary repositories for coreax
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Stencil computations in JAX☆66Updated last year
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- Minimal Implementation of Bayesian Optimization in JAX☆84Updated 6 months ago
- Fast k-NN graph construction for slow metrics☆59Updated 2 years ago
- JAX Arrays for human consumption☆88Updated last year
- Stainless neural networks in JAX☆31Updated last month
- ☆110Updated 3 weeks ago
- Neural Networks for JAX☆83Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- Mathematical operations for JAX pytrees☆189Updated 6 months ago
- ☆77Updated this week
- Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.☆58Updated this week
- Community-sourced list of papers and resources on neural simulation-based inference.☆96Updated 5 months ago
- Named Tensors for Legible Deep Learning in JAX☆152Updated this week
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 4 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- State of the art inference for your bayesian models.☆176Updated last week
- Stochastic trace estimation using JAX☆12Updated last month
- Tutorials and sampling algorithm comparisons☆69Updated this week
- Exponential families for JAX☆55Updated this week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆20Updated 2 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 3 weeks ago
- Tools for an Aesara-based PPL.☆65Updated 3 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- ☆49Updated last year
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆44Updated 5 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆332Updated this week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year