Reverse Dependencies of jax
The following projects have a declared dependency on 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
- JaxSSO — A framework for structural shape optimization based on automatic differentiation (AD) and the adjoint method, enabled by JAX
- jaxtorch — A jax based nn library
- 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
- jeometric — Graph Neural Networks in JAX
- jimGW — Gravitatioanl wave data analysis tool in Jax
- jimm — JAX Image Models
- jinns — Physics Informed Neural Network with JAX
- jit-env — A Jax interface for Reinforcement Learning environments.
- jmp — JMP is a Mixed Precision library for JAX.
- jnormcorre — Jax-accelerated implementation of normcorre
- jo3mnist — no summary
- jouissance — jouissance
- jpu — JAX + Units
- jpviz — Jaxpr Visualisation Tool
- jraph — Jraph: A library for Graph Neural Networks in Jax
- jshbtf0302 — State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch
- jsrm — Kinematic and dynamic models of continuum and articulated soft robots.
- jTWA — Truncated Wigner on GPUs based on JAX.
- jumanji — A diverse suite of scalable reinforcement learning environments in JAX
- juxai-s2 — JUX is a jax-accelerated engine for Lux-2022.
- juxai2022 — JUX is a jax-accelerated engine for Lux-2022.
- juxtapose — no summary
- jVMC — jVMC: Versatile and performant variational Monte Carlo
- jvt — Vision transformers with JAX & Flax
- kagglefold — Making protein folding accessible in kaggle platform
- kaljax — Kalman filter implemented in JAX for GPU and CPU usage
- kasearch — KA-Search: Rapid and exhaustive sequence identity search of known antibodies
- kernax — Regularized Stein thinning using JAX
- kernelbiome — A kernel-based nonparametric regression and classification framework for compositional data.
- kernex — Stencil computations in JAX.
- kfac-jax — A Jax package for approximate curvature estimation and optimization using KFAC.
- kima — A data analysis framework for exoplanet detection
- kinjax — This module calculates FK and Jacobian, accelerated by Jax to utilize JIT and GPU parallelization
- kira_llm — That's right, I'm Kira ✍️
- klujax — a KLU solver for JAX
- kornia-rs — Low level implementations for computer vision in Rust
- lagrangebench — LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
- lambeq — A QNLP toolkit
- larch6 — no summary
- last-asr — The LAttice-based Speech Transducer (LAST) library
- 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
- lineax — Linear solvers in JAX and Equinox.
- linpde-gp — Linear PDE Solvers as Gaussian Process Inference
- local-attention-flax — Local Attention - Flax Module in Jax
- localmd — Method for compressing neuroimaging data using spatially localized low-rank matrix decompositions
- long-range-models — Simple Flax implementations of long-range sequence models
- lovely-jax — 💘 Lovely JAX
- lpips-j — lpips-j – Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
- lqg — (Inverse) optimal control for linear-quadratic Gaussian systems
- lsqfitgp — A general purpose Gaussian process regression module
- madjax — differentiable matrix elements
- mapc-sim — IEEE 802.11 MAPC (C-SR) simulator
- matfree — Matrix-free numerical linear algebra.
- matrixelement — no summary
- maxspin — Estimate spatial information in spatial -omics datasets.
- 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.
- meent — no summary
- MeerKATgen — MeerKAT Radio Telescope simulation data generator
- mellon — Non-parametric density estimator.
- meloetta — no summary
- memory-efficient-attention — Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
- menzalib — Funzioni utili per lab3
- meow-sim — Modeling of Eigenmodes and Overlaps in Waveguide Structures
- metax — A collection of extensions for meta-learning in JAX
- mewtax — Differentiable minimization in jax using Newton's method.
- microsim — Light microscopy simulation in python
- miles-guess — no summary
- minimax-lib — Efficient baselines for autocurricula in JAX
- MinVectorDB — A pure Python-implemented, lightweight, server-optional, multi-end compatible, vector database deployable locally or remotely.
- mistral-jax — JAX implementation of the Mistral model.
- mistral-v0.2-jax — JAX implementation of the Mistral v0.2 base model.