Reverse Dependencies of jaxlib
The following projects have a declared dependency on jaxlib:
- jax-dips — Differentiable 3D interfacial PDE solvers written in JAX using the Neural Bootstrapping Method.
- jax-enums — JAX-compatible Enumerations.
- jax-fid — FID computation in Jax/Flax.
- jax-fixedpoint — Implicit and competitive differentiation in JAX.
- jax-fixedpoint-test-manueldelverme — Implicit and competitive differentiation in JAX.
- jax-flows — Normalizing Flows for JAX
- jax-ifnt — Execute runtime assertions, indexing checks, and more if `jax` code is not traced.
- jax-jumpy — Common backend for Jax or Numpy.
- jax-jynx — A neural network library using jax
- jax-lorax — A JAX transform which applies LoRA to arbitrary JAX functions/models
- jax-md — Differentiable, Hardware Accelerated, Molecular Dynamics
- jax-metrics — no summary
- jax-nerf — Jax implementation of neural radiance fields
- jax-quant-finance — High Performance Quantative Finance Library on JAX
- jax-quantity — Quantities in JAX
- jax-random-projections — sklearn's random projection with JAX to run on a GPU
- jax-resnet — Framework-agnostic library for checking array shapes at runtime.
- jax-russell — Financial options valuations and auto-diffed risk measures ("greeks" in Jax.
- jax-scipy-spatial — Scipy spatial API for JAX
- jax-sgmc — Stochastic Gradient Monte Carlo in Jax
- jax-sph — JAX-SPH: Smoothed Particle Hydrodynamics in JAX
- jax-toolkit — A collection of jax functions to help with common machine/deep learning related functionality.
- jax-verify — A library for neural network verification.
- jaxampler — A JAX-based statistical sampling toolkit
- jaxbind — Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
- jaxbook — Iterative algorithms for numerical linear algebra with JAX
- jaxdifferentialgeometry — Differential geometry using jax
- jaxDiversity — jax implementation for metalearning neuronal diversity
- jaxdsp — Fast, differentiable audio processors on the CPU or GPU, with a browser client for real-time control
- jaxex — A tool for creating science experiments in jax, torch, brax, etc
- JaxFin — JaxFin is a powerful and versatile Python library designed for pricing exotic options using a range of advanced financial techniques.
- jaxfit — Uses JAX for GPU/TPU accelerated nonlinear least-squares curve fitting
- jaxformers — 'Attention is all you need' in JAX (Flax)
- jaxGW — Gravitatioanl wave data analysis tool in Jax
- jaxio — Input pipelines for JAX, in JAX
- jaxite — A Homomorphic Encryption implementation (CGGI) written in JAX
- JaxKern — Kernels in Jax.
- jaxkern-nightly — Kernels in Jax.
- jaxknife — no summary
- jaxlinop-nightly — A JaxLinOp library.
- jaxmarl — Multi-Agent Reinforcement Learning with JAX
- JaxNRSur — Numerical relativity surrogate model for gravitational waveforms in Jax
- jaxns — Nested Sampling in JAX
- jaxonfunctions — Functions for JAX
- jaxonloader — A dataloader, but for JAX/Numpy
- jaxonmodels — JAX models for deep learning
- jaxonnxruntime — Jaxonnxruntime: JAX based ONNX Runtime.
- jaxoplanet — Probabilistic models for exoplanet inference in JAX
- jaxopt — Hardware accelerated, batchable and differentiable optimizers in JAX.
- jaxparrow — Computes the inversion of the cyclogeostrophic balance based on a variational formulation approach, using JAX
- jaxprvis — A visualizer for JaxPrs
- jaxqualin — A package for extracting quasinormal modes from time domain data
- jaxrenderer — Jax implementation of rasterizer renderer.
- jaxrie — Riemannian JAX
- jaxsim — A differentiable physics engine and multibody dynamics library for control and robot learning.
- jaxsnn — jaxsnn is an event-based approach to machine-learning-inspired training and simulation of SNNs, including support for neuromorphic backends (BrainScaleS-2).
- jaxson — A generative art libary based on Jax
- jaxtpu — A helper package to install the latest JAX on TPUs, along with all necessary dependencies (e.g. libtpu-nightly)
- JaxUtils — Utility functions for JaxGaussianProcesses
- jaxutils-nightly — Utility functions for JaxGaussianProcesses
- jaxwell — Jaxwell is JAX + Maxwell: an iterative solver for solving the finite-difference frequency-domain Maxwell equations on NVIDIA GPUs
- jaxwt — Differentiable and gpu enabled fast wavelet transforms in JAX
- jimGW — Gravitatioanl wave data analysis tool in Jax
- jmp — JMP is a Mixed Precision library for JAX.
- jnormcorre — Jax-accelerated implementation of normcorre
- jouissance — jouissance
- jpu — JAX + Units
- jraph — Jraph: A library for Graph Neural Networks in Jax
- jshbtf0302 — State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch
- jsmfsb — Python+JAX code relating to the textbook, Stochastic modelling for systems biology, third edition
- jTWA — Truncated Wigner on GPUs based on JAX.
- juxtapose — no summary
- jVMC — jVMC: Versatile and performant variational Monte Carlo
- jvt — Vision transformers with JAX & Flax
- kasearch — KA-Search: Rapid and exhaustive sequence identity search of known antibodies
- kernelbiome — A kernel-based nonparametric regression and classification framework for compositional data.
- kfac-jax — A Jax package for approximate curvature estimation and optimization using KFAC.
- kima — A data analysis framework for exoplanet detection
- kira_llm — That's right, I'm Kira ✍️
- klujax — a KLU solver for JAX
- lagrangebench — LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
- lambeq — A QNLP toolkit
- lazyqml — LazyQML benchmarking utility to test quantum machine learning models.
- learned-optimization — Train learned optimizers in Jax.
- lenscharm — A Charming, Bayesian, Strong Lensing Framework
- levy-stable-jax — Implementation of the Lévy alpha-stable distributions for Jax.
- liesel — A probabilistic programming framework with a focus on semi-parametric regression
- lightcurver — A thorough structure for precise photometry and deconvolution of time series of wide field images.
- lightning-thunder — Lightning Thunder project.
- lightweight-mmm — Package for Media-Mix-Modelling
- local-attention-flax — Local Attention - Flax Module in Jax
- localmd — Method for compressing neuroimaging data using spatially localized low-rank matrix decompositions
- lsqfitgp — A general purpose Gaussian process regression module
- madjax — differentiable matrix elements
- mapc-sim — IEEE 802.11 MAPC (C-SR) simulator
- matrixelement — no summary
- MCCube — Markov chain cubature via JAX.
- mclmc — Faster gradient based sampling
- mctx — Monte Carlo tree search in JAX.
- MDSuite — A postprocessing tool for molecular dynamics simulations.