Reverse Dependencies of jaxlib

The following projects have a declared dependency on jaxlib:

• GridPolator — Interpolate a grid of spectroscopic models.
• gsmvi — Implementation of Gaussian score matching for variational inference (arXiv:2307.07849)
• gunpowder — A library to facilitate machine learning on large, multi-dimensional images.
• gwkokab — A JAX-based gravitational-wave population inference
• gwpopulation — Unified population inference
• gymnasium — A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym).
• gymnax — JAX-compatible version of Open AI's gym environments
• halospec — Halo Spectroscopy for JWST/MIRI
• hamux — A Deep Learning framework built around ENERGY
• harmonic — Python package for efficient Bayesian evidence computation
• harmonix — Analytic interferometry of stellar surfaces using spherical harmonics in Jax
• harpy-analysis — single-cell spatial proteomics analysis that makes you happy
• hbmep — Hierarchical Bayesian estimation of motor-evoked potential (MEP) size recruitment curves
• helax — Python package for computing helicity amplitudes
• hgan — A package to infer interpretable dynamics from images of a mechanical system.
• hnoca — Human Neural Organoid Cell Atlas Toolbox
• Hoi — Higher Order Interactions
• hotaru — High performance Optimizer to extract spike Timing And cell location from calcium imaging data via lineaR impUlse
• hxrate — HX Rate Fitting
• hybridq — Hybrid Simulator for Quantum Circuits
• hypervecs — no summary
• IFE-Surrogate — A brief description of your library
• igenerator — Generating random numbers faster than numpy
• ikpls — Improved Kernel PLS and Fast Cross-Validation.
• imax — Image augmentation library for Jax
• iminuit — Jupyter-friendly Python frontend for MINUIT2 in C++
• IMNN — Using neural networks to extract sufficient statistics from data by maximising the Fisher information
• impt — Auto-diff Estimator of Lensing Perturbations
• imt-tree_utils — Utilities for working with Pytrees
• infnum — Infinite and infinitesimal numbers for JAX
• invrs-gym — A collection of inverse design challenges
• invrs-opt — Algorithms for inverse design
• jaims — Library for jax based affine-invariant MCMC sampling
• jammer — Library for jax based affine-invariant MCMC sampling
• janus-sim — A JAX Neural Simulator
• jars — Next-generation objects for omics data
• javiche — A JAX wrapper around ceviche to make interoperability easier. In the future it might make sense to update ceviche itself to use JAX internally.
• jax — Differentiate, compile, and transform Numpy code.
• jax-am — GPU-accelerated simulation toolbox for additive manufacturing based on JAX.
• jax-cgd — A package for competitive gradient descent and its adaptive version using JAX
• jax-data — Native data handling for JAX
• jax-dataclasses — Dataclasses + JAX
• jax-ddp — no summary
• jax-dimenet — DimeNet++ in Jax.
• 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-flash-attn2 — Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attention with support for different backends (GPU, TPU, CPU) and platforms (Triton, Pallas, 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-scalify — JAX Scalify: end-to-end scaled arithmetic.
• 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
• jaximal — A JAX-based PyTree manipulation library
• 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
• jaxlayerlumos — A JAX-based efficient transfer-matrix method framework for optical simulations
• 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