Reverse Dependencies of jax

The following projects have a declared dependency on 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
• 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
• 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
• 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.
• 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
• 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
• craynn — Yet another neural network toolkit.
• crayopt — Yet another optimization toolkit for jax.
• crikit — Constitutive Relation Inference Toolkit
• crispio — Designing CRISPR-Cas guide RNAs in bacteria.
• crossfit — Offline inference and metric calculation library
• cryojax — Cryo-EM image simulation and analysis powered by JAX
• cuequivariance-jax — CUDA accelerated equivariant operations
• cupbearer — A library for mechanistic anomaly detection
• d3p — Differentially-Private Probabilistic Programming using NumPyro and the differentially-private variational inference algorithm
• dadapy — Distance-based Analysis of DAta-manifolds in python
• dalle-mini — DALL·E mini - Generate images from a text prompt
• dandeliion-client — Client for accessing Dandeliion services
• danling — Scaffold for experienced Machine Learning Researchers
• data-decomposition-platform — Data Decomposition Platform library for data disaggregation.
• dataclass-array — Dataclasses that behave like numpy arrays (with indexing, slicing, vectorization).
• datasets — HuggingFace community-driven open-source library of datasets
• datasets-fast — Optimisations for HuggingFace Datasets with array features
• decayangle — A tool for wigner rotations in n-body decays
• declearn — A modular and extensible framework for Federated Learning.
• decoding — Composable inference algorithms with LLMs and programmable logic
• deepchem — Deep learning models for drug discovery, quantum chemistry, and the life sciences.
• deeperwin — A JAX based framework to calculate deep-learning VMC solutions to the Schrödinger equation
• deephyper — Scalable asynchronous neural architecture and hyperparameter search for deep neural networks
• deepmd-kit — A deep learning package for many-body potential energy representation and molecular dynamics
• deepqmc — Deep-learning quantum Monte Carlo for electrons in real space
• deluca — no summary
• dendritex — Dendrite Modeling in Python
• desc-opt — Computes, analyzes and optimizes 3D MHD equilibria for stellarators and tokamaks
• device-picker — A device picker for machine learning frameworks like PyTorch, TensorFlow, and JAX.
• dgmrf — Deep Gaussian Markov Random Fields and their extensions
• dibs-lib — DiBS: Differentiable Bayesian Structure Learning
• diffaaable — JAX-differentiable AAA algorithm
• diffeqzoo — Ordinary differential equation zoo