Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• 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
• braincell — Biologically Detailed Brain Cell Modeling in JAX
• braincore — The Core System for General-purpose Brain Dynamics Programming Framework.
• brainevent — Event-driven Computation in JAX for Brain Dynamics.
• 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
• brainsolver — Reversible Solver for Brain Modeling
• 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 for General-Purpose Brain Dynamics Modeling
• brax — A differentiable physics engine written in JAX.
• brax-jumpy — Common backend for JAX or numpy.
• breedgym — Suite of Gymnasium environments for optimizing breeding programs
• brioche-enrichment — Bayesian tests for set enrichment.
• bstaple — Bayesian version of STAPLE
• BSTPP — Bayesian Spatiotemporal Point Process
• bsts — Python library for Bayesian structural time series
• budoux — BudouX is the successor of Budou
• cambrian-core — Artificial Cambrian Intelligence
• capfinder — A package for decoding RNA cap types
• capibara-ent — A flexible multimodal AI library for advanced contextual understanding and deployment.
• capibara-model — A State Space and Mamba-based language model framework
• cardio-rl — Cardio RL. In development...
• cascades — no summary
• casus — no summary
• catenets — Conditional Average Treatment Effect Estimation Using Neural Networks
• caujax — Causal Jax
• causal-discovery — no summary
• cax — Cellular Automata Accelerated in JAX
• cbfax — cbf with jax
• cbfpy — Control Barrier Functions in Python
• cca-zoo — Canonical Correlation Analysis Zoo: A collection of Regularized, Deep Learning based, Kernel, and Probabilistic methods in a scikit-learn style framework
• celerite2 — Fast and scalable Gaussian Processes in 1D
• celldisect — Cell DISentangled Experts for Covariate counTerfactuals (CellDISECT). Causal generative model designed to disentangle known covariate variations from unknown ones at test time while simultaneously learning to make counterfactual predictions.
• cellrank — CellRank: dynamics from multi-view single-cell data
• ceml — Counterfactuals for explaining machine learning models - A Python toolbox
• cfnet — A counterfactual explanation library using Jax
• cfrx — Counterfactual Regret Minimization in Jax
• chemise — Wrapper for training flax models
• chemtrain — Machine-learning deep molecular dynamics potentials.
• chex — Chex: Testing made fun, in JAX!
• chimera-gw — Combined Hierarchical Inference Model for Electromagnetic and gRavitational-wave Analysis
• chromatix — Differentiable computational optics library using JAX!
• chromax — Breeding simulator based on JAX
• ciclo — no summary
• cleanrl — High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features
• clearbox-synthetic-kit — Clearbox AI's all-in-one solution for generation and evaluation of synthetic tabular and time-series data.
• cleverhans — no summary
• clip-jax — Training of CLIP in JAX
• clu — Set of libraries for ML training loops in JAX.
• cmonge — Extension of the Monge Gap to learn conditional optimal transport maps
• coax — Plug-n-play reinforcement learning with Gymnasium and JAX
• coaxtract — Permittivity extraction from open-ended coaxial probe measurements
• codex-africanus — Radio Astronomy Building Blocks
• cody-adapter-transformers — A friendly fork of HuggingFace's Transformers, adding Adapters to PyTorch language models
• coffeegrinder — COFFEE is an MPI-Parallelised Python package for the numerical evolution of time-dependent differential equations.
• coix — Inference Combinators in JAX
• colabfold — Making protein folding accessible to all. Predict proteins structures both in google colab and on your machine
• coltra — Coltra is a simple moddable RL algorithm implementation
• coltra-rl — Coltra-RL is a simple moddable RL algorithm implementation
• CombineHarvesterFlow — Harvest and combine chains from different data sets
• como-ebnm — no summary
• competitionassay — Software for the analysis of competition assay data.
• computegraph — computegraph is a tool for managing computational graphs using networkx
• conformal-region-designer — Creates intelligently shaped conformal prediction regions that respect multimodality and convexity.
• connectomics — Common repository for connectomics-related libraries.
• connex — Fine-grained, dynamic control of neural network topology in JAX.
• continuation-jax — Continuation Methods for Deep Neural Networks.
• coola — Library to check equality between two complex/nested objects
• coordinax — Coordinates in JAX
• coppafish — coppaFISH software for Python
• coreax — Jax coreset algorithms.
• correctionlib-gradients — Automatic differentiation for high-energy physics correction factor calculations.
• cortecs — Compress opacity for radiative transfer
• cosmax — Fast and differentiable implementations of operations needed for inference and analysis in cosmology. Powered by JAX
• cosmologix — Cosmological distance computations in FLRW universe using JAX
• covvfit — no summary
• cpa-tools — Compositional Perturbation Autoencoder (CPA)
• cpprb — ReplayBuffer for Reinforcement Learning written by C++ and Cython
• cpwxtract — Permittivity extraction from open-ended coaxial probe measurements
• cr-nimble — Iterative algorithms for numerical linear algebra with JAX
• craftax — An open-world environment for training RL agents
• craftaxlm — no summary
• craftground — Lightweight Minecraft Environment for Reinforcement Learning
• craynn — Yet another neural network toolkit.
• crayopt — Yet another optimization toolkit for jax.
• cricket-tannmay — A computer vision package to recognize and assess different cricket shots and bowling videos.
• cricket-tannmay-headless — A computer vision package to recognize and assess different cricket shots and bowling videos.
• crikit — Constitutive Relation Inference Toolkit
• crispio — Designing CRISPR-Cas guide RNAs in bacteria.
• crossfit — Offline inference and metric calculation library