Reverse Dependencies of jaxlib
The following projects have a declared dependency on jaxlib:
- 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
- agjax — A jax wrapper for autograd-differentiable functions.
- ai2-tango — A library for choreographing your machine learning research.
- ajents — RL agents in JAX
- 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.
- aqtp — Accurate Quantized Training library.
- ARLBench — Python Boilerplate that contains all the code you need to create a Python package.
- array-api-jax-compat — Array-API JAX compatibility
- ars-dev — A JAX/numpy implementation of Adaptive Rejection Sampling (ARS) introduced by Gilks and Wild in 1992
- astax — A Jax based neural network library for research
- auto-uncertainties — Linear Uncertainty Propagation with Auto-Differentiation
- 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
- axlearn — AXLearn
- bartz — A JAX implementation of BART
- BaSiCpy — A python package for background and shading correction of optical microscopy images
- 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
- bayex — Minimal Bayesian Optimization Implementation with Gaussian Processes written in JAX.
- bbo-calibcam — Calibrate intrinsic and extrinsic parameters of cameras with charuco boards
- benchmark-mi — Estimators of mutual information and distributions used to benchmark them.
- bio-embeddings — A pipeline for protein embedding generation and visualization
- 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
- boax — Boax is a Bayesian Optimization library for JAX.
- 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.
- brainlib — Scalable Online Learning in Biological Neural Networks.
- 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
- brainscale — The Scalable Online Learning Algorithm and Framework for Biological Spiking Neural Networks.
- 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
- brax — A differentiable physics engine written in JAX.
- brax-jumpy — Common backend for JAX or numpy.
- BSTPP — Bayesian Spatiotemporal Point Process
- casus — no summary
- catenets — Conditional Average Treatment Effect Estimation Using Neural Networks
- causal-discovery — no summary
- cbfpy — Control Barrier Functions in Python
- 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.
- ceml — Counterfactuals for explaining machine learning models - A Python toolbox
- chemtrain — Machine-learning deep molecular dynamics potentials.
- chex — Chex: Testing made fun, in 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
- 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
- coltra — Coltra is a simple moddable RL algorithm implementation
- coltra-rl — Coltra-RL is a simple moddable RL algorithm implementation
- computegraph — computegraph is a tool for managing computational graphs using networkx
- coola — Library to check equality between two complex/nested objects
- coordinax — Coordinates in JAX
- coppafish — coppaFISH software for Python
- correctionlib-gradients — Automatic differentiation for high-energy physics correction factor calculations.
- cortecs — Compress opacity for radiative transfer
- covvfit — no summary
- cpa-tools — Compositional Perturbation Autoencoder (CPA)
- cpwxtract — Permittivity extraction from open-ended coaxial probe measurements
- cr-nimble — Iterative algorithms for numerical linear algebra with JAX
- crikit — Constitutive Relation Inference Toolkit
- cryojax — Cryo-EM image simulation and analysis powered by JAX
- dadapy — Distance-based Analysis of DAta-manifolds in python
- 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
- 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
- deepqmc — Deep-learning quantum Monte Carlo for electrons in real space
- dendritex — Dendrite Modeling in Python