Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• absl-extra — A wrapper to run and monitor absl app.
• accurating — AccuRating is a library for accurate player ranking based on match results.
• acqdp — Alibaba Cloud Quantum Development Platform
• adam-core — Core libraries for the ADAM platform
• adam-robotics — Automatic Differentiation for rigid-body-dynamics AlgorithMs
• adastra — Astra - Pytorch
• adirondax — A differentiable astrophysics simulator in JAX
• advanced-global-optimizers — A package that integrates many advanced global optimizers
• aex — Sampling with Blackjax on Aesara
• agilerl — AgileRL is a deep reinforcement learning library focused on improving RL development through RLOps.
• agjax — A jax wrapper for autograd-differentiable functions.
• ai-edge-torch-nightly — Supporting PyTorch models with the Google AI Edge TFLite runtime.
• ai-helpers-jax-utils — Common jax utils
• ai-hexagon — An objective way to evaluate neural network architectures
• ai2-tango — A library for choreographing your machine learning research.
• aigs — no summary
• aironsuit — A model wrapper for automatic model design and visualization purposes.
• ajents — RL agents in JAX
• alpa — Alpa automatically parallelizes large tensor computation graphs and runs them on a distributed cluster.
• alphafold — An implementation of the inference pipeline of AlphaFold v2.0.This is a completely new model that was entered as AlphaFold2 in CASP14 and published in Nature.
• alphafold-colabfold — An implementation of the inference pipeline of AlphaFold v2.3.1. This is a completely new model that was entered as AlphaFold2 in CASP14 and published in Nature. This package contains patches for colabfold.
• alphafold-kagglefold — An implementation of the inference pipeline of AlphaFold v2.0.This is a completely new model that was entered as AlphaFold2 in CASP14 and published in Nature. This package contains patches for kagglefold
• alquimodelia — Keras based model builder
• alquitable — Keras-core based tools to enhance Alquimodelia
• anabel — An end to end differentiable finite element framework.
• anacal — no summary
• ananke-causal — Ananke, named for the Greek primordial goddess of necessity and causality, is a python package for causal inference using the language of graphical models.
• angmom-suite — A package for working with phenomenological spin and angular momentum operators
• annax — Fast and memory-efficient approximate nearest neighbor search with JAX
• anon — Research workspace.
• anyboxes — Lightweight package for managing bounding boxes that works seamlessly with most computing frameworks.
• apax — Atomistic Learned Potential Package in JAX
• apebench — Benchmark suite for Autoregressive Neural Emulators of PDEs in JAX.
• apollo-toolbox-py — no summary
• appletree — A high-Performance Program simuLatEs and fiTs REsponse of xEnon.
• apropos-ai — Learning algorithms for production language model programs
• aqsc — Constructs QS stellarator equilibrium to all orders.
• aqtp — Accurate Quantized Training library.
• ARLBench — Python Boilerplate that contains all the code you need to create a Python package.
• array-api-compat — A wrapper around NumPy and other array libraries to make them compatible with the Array API standard
• array-api-jax-compat — Array-API JAX compatibility
• array-typing — Add your description here
• ars-dev — A JAX/numpy implementation of Adaptive Rejection Sampling (ARS) introduced by Gilks and Wild in 1992
• arxette — autoregressive fit
• aspcol — no summary
• astax — A Jax based neural network library for research
• AttentionGrid — AttentionGrid - Library
• augmax — Efficiently Composable Data Augmentation on the GPU with Jax
• auto-uncertainties — Linear Uncertainty Propagation with Auto-Differentiation
• autobound — no summary
• autodiscjax — python library built on top of jax to facilitate automated exploration and simulation of computational models of biological processes
• autodyn — Differentiable Dynamical Systems
• AutoEis — A tool for automated EIS analysis by proposing statistically plausible ECMs.
• autopdex — An automized PDE solver for accelerators based on JAX
• avici — Amortized Inference for Causal Structure Learning
• awblib — A bunch of Automatic White-Balancing (AWB) Algorithm implementations
• aws-fortuna — A Library for Uncertainty Quantification.
• axlearn — AXLearn
• bartz — A JAX implementation of BART
• BaSiCpy — A python package for background and shading correction of optical microscopy images
• bax — A flexible trainer interface for Jax and Haiku.
• bayes-jones — Bayesian inference of Jones matrices.
• bayes_spec — A Bayesian Spectral Line Modeling Framework for Astrophysics
• BayesComBat — Fully Bayesian ComBat Harmonization
• bayesian-models — A package for building common bayesian models in pymc
• BayesMBAR — Add your description here
• bayesn — Hierarchical Bayesian modelling of type Ia SNe
• bayesnf — Scalable spatiotemporal prediction with Bayesian neural fields
• bayeux-ml — Stitching together probabilistic models and inference.
• bayex — Minimal Bayesian Optimization Implementation with Gaussian Processes written in JAX.
• bbhamux — Minimal library to construct Hierarchical Associative Memories
• bbo-calibcam — Calibrate intrinsic and extrinsic parameters of cameras with charuco boards
• beartype — Unbearably fast near-real-time hybrid runtime-static type-checking in pure Python.
• benchmark-mi — Estimators of mutual information and distributions used to benchmark them.
• benchmarx — Tools for benchmarking optimization methods
• bfastor — Fast cryo-EM B-factor refinement
• bfbrain — Use active learning to determine the bounded-from-below region in parameter space of a multiscalar potential in quantum field theory.
• bilby-numpyro — no summary
• biofit — BioFit: Bioinformatics Machine Learning Framework
• biosets — Bioinformatics datasets and tools
• blackjax — Flexible and fast sampling in Python
• blackjax-nightly — Flexible and fast sampling in Python
• bm25s — An ultra-fast implementation of BM25 based on sparse matrices.
• bmr4pml — Bayesian model reduction for probabilistic machine learning
• boax — Boax is a Bayesian Optimization library for JAX.
• bobbin — Tools for making training loops with flax.linen models.
• bojaxns — Bayesian Optimisation with JAXNS
• bottleneck-transformer-flax — Bottleneck Transformer - Flax
• bqskit-qfactor-jax — QFactor GPU implementation in BQSKit using JAX
• Brain-Py — BrainPy: Brain Dynamics Programming in Python
• braincore — The Core System for General-purpose Brain Dynamics Programming Framework.
• brainos — Brain Operating System
• brainpy — BrainPy: Brain Dynamics Programming in Python
• brainpy.core — The core system for BrainPy programming framework.
• brainpy-largescale — brainpy-largescale depends on brainpy
• brainpylib-test — C++/CUDA Library for BrainPy
• brainstate — A ``State``-based Transformation System for Program Compilation and Augmentation.
• braintaichi — Leveraging Taichi Lang to customize brain dynamics operators
• braintools — The Toolbox for Brain Dynamics Programming.
• brainunit — Physical Units and Unit-aware Mathematical System in JAX