A library for coreset algorithms, written in JAX for fast execution and GPU support.
☆37Feb 26, 2026Updated this week
Alternatives and similar repositories for coreax
Users that are interested in coreax are comparing it to the libraries listed below
Sorting:
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 4 months ago
- A nascent Jax-based package for virtual brain modeling.☆12Feb 3, 2026Updated last month
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- ☆219Feb 21, 2026Updated last week
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10May 7, 2024Updated last year
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 6 months ago
- ☆19Jan 25, 2026Updated last month
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 4 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Jul 23, 2023Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated last month
- Lectures for the 2019 Kavli Summer Program in Astrophysics☆17Mar 11, 2020Updated 5 years ago
- Coordinates in JAX☆29Jan 31, 2026Updated last month
- Coordinate Axes for JAX☆62Updated this week
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Bayesian Dynamic Linear Models for time-series analysis☆26Feb 18, 2022Updated 4 years ago
- Orthogonal polynomials with JAX☆28Jan 20, 2026Updated last month
- Minimal Implementation of Bayesian Optimization in JAX☆104Apr 24, 2025Updated 10 months ago
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 7 months ago
- ☆22May 6, 2020Updated 5 years ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆111Apr 7, 2025Updated 10 months ago
- Group Iterative Multiple Model Estimation - R version☆29Nov 12, 2025Updated 3 months ago
- Spiking neural networks in JAX with equinox.☆29Aug 29, 2024Updated last year
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Feb 22, 2026Updated last week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆43Updated this week
- Signax: Signature computation in JAX☆38Jan 27, 2025Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Feb 23, 2025Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆32May 12, 2023Updated 2 years ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- A collection of graph neural networks implementations in JAX☆35Nov 28, 2023Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Feb 12, 2026Updated 2 weeks ago
- Interpolation and function approximation with JAX☆250Feb 16, 2026Updated 2 weeks ago
- Tutorials and sampling algorithm comparisons☆80Updated this week
- Gaussian processes in JAX and Flax.☆596Updated this week
- State of the art inference for your bayesian models.☆233Jan 14, 2026Updated last month
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago