PhilipVinc / numba4jaxView external linksLinks
Use numba-compiled kernels from within Jax
☆31May 1, 2025Updated 9 months ago
Alternatives and similar repositories for numba4jax
Users that are interested in numba4jax are comparing it to the libraries listed below
Sorting:
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX☆18Feb 6, 2026Updated last week
- PyBlastAfterglow: fast modeling of GRB and Kilonova afterglows☆11Jan 16, 2025Updated last year
- convenient callbacks to use when you sample in Turing.jl.☆15Dec 30, 2025Updated last month
- Configuration Space Covariances for Projected Galaxy 2-Point Statistics☆14Aug 13, 2025Updated 6 months ago
- JAX + Units☆52Dec 15, 2025Updated 2 months ago
- High-performance C++ framework for afterglow modeling with Python bindings. Features fast multi-wavelength light curve generation, MCMC p…☆36Feb 9, 2026Updated last week
- A web portal that offers more features and a better user interface for reading eprints☆25Jun 16, 2016Updated 9 years ago
- Documentation:☆128May 22, 2023Updated 2 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week
- Estimate the autocorrelation time of time-series data very quickly☆36Jun 18, 2023Updated 2 years ago
- An augmented version of cProfile and Snakeviz☆27Aug 30, 2017Updated 8 years ago
- A web interface for exploring PyMC3 traces☆46Feb 14, 2018Updated 8 years ago
- Flexible, general-purpose VMC framework, built on JAX.☆29Nov 25, 2025Updated 2 months ago
- This repo plans to provide a low-level Julia wrapper for BLIS typed interface.☆26Oct 15, 2025Updated 4 months ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week
- OpTree: Optimized PyTree Utilities☆205Feb 4, 2026Updated last week
- ☆12Oct 7, 2020Updated 5 years ago
- ✨🛰 Fast and flexible spectral fitting in Julia.☆20Jan 10, 2026Updated last month
- Compose computational workflows from Python☆11Oct 9, 2024Updated last year
- ☆28Jan 12, 2022Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- A light interface to serial and multi-threaded Sequential Monte Carlo☆32Dec 15, 2021Updated 4 years ago
- LoRA for arbitrary JAX models and functions☆145Feb 26, 2024Updated last year
- Stencil computations in JAX☆70Nov 15, 2025Updated 3 months ago
- Distributed table structures and data manipulation operations built on top of Dagger.jl☆79Apr 8, 2024Updated last year
- Probabilistic Programming and Nested sampling in JAX☆221Dec 15, 2025Updated 2 months ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago
- Auto differentiation over linear algebras (a Zygote extension)☆35Nov 24, 2025Updated 2 months ago
- Neural-Network representation of Quantum Systems☆41Apr 29, 2020Updated 5 years ago
- Revisiting Whittaker-Henderson Smoothing☆11Jun 19, 2025Updated 7 months ago
- Sparse symmetric indefinite solver implemented with a runtime system☆13May 11, 2020Updated 5 years ago
- Efficient implementation of a parallel bucket-sort with OpenMP☆10Mar 19, 2017Updated 8 years ago
- ☆11Dec 4, 2024Updated last year
- Quantnet: SFE quantlets☆11Oct 27, 2025Updated 3 months ago
- Numpy compatible polynomial representation☆12Jan 2, 2026Updated last month
- Interstellar Magnetic Field Inference Engine, a modular open source framework for doing inference on generic parametric models of the Gal…☆12Jul 20, 2023Updated 2 years ago
- Incremental 3D Delaunay Tetrahedralization☆14Jul 21, 2024Updated last year