Tidy autoregressive inference in JAX
☆15Sep 1, 2025Updated last year
Alternatives and similar repositories for scanagram
Users that are interested in scanagram are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Parallel hyperparameter tuning with JAX☆40Jul 18, 2026Updated last month
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- ☆68Aug 21, 2024Updated 2 years ago
- Stencil computations in JAX☆73Nov 15, 2025Updated 9 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- LoRA for arbitrary JAX models and functions☆143Feb 26, 2024Updated 2 years ago
- Implementation of Forward Laplacian algorithm in JAX☆103Updated this week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆60May 19, 2026Updated 3 months ago
- JAX Arrays for human consumption☆113Oct 23, 2025Updated 10 months ago
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆24Apr 17, 2024Updated 2 years ago
- Automatic Differentiation for high-performance stencil loops☆13Mar 25, 2021Updated 5 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 6 years ago
- Jax Decompiler☆24Apr 22, 2025Updated last year
- ☆18Mar 20, 2022Updated 4 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- ☆311Updated this week
- Spiking neural networks in JAX with equinox.☆29Aug 29, 2024Updated 2 years ago
- JAX for Graphcore IPU (experimental)☆21Mar 12, 2024Updated 2 years ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- Coordinates in JAX☆37Updated this week
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆329Updated this week
- ☆16Jun 3, 2025Updated last year
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆82Feb 12, 2026Updated 6 months ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated 2 years ago
- Coordinate Axes for JAX☆67Aug 13, 2026Updated 3 weeks ago
- Data compression in JAX☆67Nov 23, 2024Updated last year
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- AutoBound automatically computes upper and lower bounds on functions.☆365Oct 24, 2025Updated 10 months ago
- Procedural Generation Using Differentiable Programming☆16Sep 29, 2019Updated 6 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆326Aug 6, 2026Updated 3 weeks ago
- Orthogonal polynomials with JAX☆30Jul 20, 2026Updated last month
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆22Jul 20, 2024Updated 2 years ago
- Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX☆22Jun 24, 2026Updated 2 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆166Oct 18, 2024Updated last year
- This repo contains the code associated to the paper: "Constrained Causal Bayesian Optimization" by Aglietti Virginia, Alan Malek, Ira Kt…☆16Jun 18, 2024Updated 2 years ago
- Fast, simple, cryptographically strong random numbers in C++. Experimental.☆19Dec 12, 2013Updated 12 years ago
- Codebase for Obfuscated Activations Bypass LLM Latent-Space Defenses☆32Feb 11, 2025Updated last year
- This is the code corresponding to our publication introducing ConvDecoder with physics-based regularization (CD+r) for MRI☆10Feb 6, 2026Updated 6 months ago