Use numba-compiled kernels from within Jax
☆32May 1, 2025Updated 10 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
- PyBlastAfterglow: fast modeling of GRB and Kilonova afterglows☆12Jan 16, 2025Updated last year
- convenient callbacks to use when you sample in Turing.jl.☆15Feb 22, 2026Updated 2 weeks ago
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- JAX + Units☆52Mar 2, 2026Updated last week
- A web portal that offers more features and a better user interface for reading eprints☆25Jun 16, 2016Updated 9 years ago
- Subsumed into xnd☆81Nov 17, 2023Updated 2 years ago
- High-performance C++ framework for afterglow modeling with Python bindings. Features fast multi-wavelength light curve generation, MCMC p…☆37Mar 3, 2026Updated last week
- Documentation:☆129May 22, 2023Updated 2 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆517Mar 2, 2026Updated last 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
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated 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
- Flexible, general-purpose VMC framework, built on JAX.☆29Nov 25, 2025Updated 3 months ago
- OpTree: Optimized PyTree Utilities☆205Mar 2, 2026Updated last week
- Modular framework for Reinforcement Learning in python☆183Feb 1, 2023Updated 3 years ago
- ☆28Jan 12, 2022Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- ✨🛰 Fast and flexible spectral fitting in Julia.☆20Jan 10, 2026Updated last month
- 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 2 years ago
- Probabilistic Programming and Nested sampling in JAX☆223Feb 27, 2026Updated last week
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- A tool used for calculating the expected secondary emission from DM annihilation and placing constraints on dark matter properties, incor…☆14Jun 9, 2022Updated 3 years 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
- Neural-Network representation of Quantum Systems☆41Apr 29, 2020Updated 5 years ago
- Auto differentiation over linear algebras (a Zygote extension)☆35Nov 24, 2025Updated 3 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆27Oct 13, 2022Updated 3 years ago
- Reborn cscope extension for emacs☆10Apr 21, 2024Updated last year
- Numpy compatible polynomial representation☆12Jan 2, 2026Updated 2 months ago
- Revisiting Whittaker-Henderson Smoothing☆11Jun 19, 2025Updated 8 months ago
- A formalization of (homotopy) limits in Homotopy Type Theory☆11Jan 11, 2015Updated 11 years ago
- ☆11Dec 4, 2024Updated last year
- multiplicatively convolutional fast integral transforms, implementing FFTLog☆34Feb 16, 2025Updated last year
- ☆11Jun 20, 2022Updated 3 years ago
- Quantnet: SFE quantlets☆11Oct 27, 2025Updated 4 months ago