Reverse Dependencies of jax
The following projects have a declared dependency on jax:
- clip-jax — Training of CLIP in JAX
- clu — Set of libraries for ML training loops in JAX.
- 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
- 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 — A library to check if two complex/nested objects are equal or not
- coordinax — Vectors 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
- 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
- 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.
- cryojax — Cryo-EM image simulation and analysis powered by JAX
- cupbearer — A library for mechanistic anomaly detection
- d3p — Differentially-Private Probabilistic Programming using NumPyro and the differentially-private variational inference algorithm
- dalle-mini — DALL·E mini - Generate images from a text prompt
- 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
- decayangle — A tool for wigner rotations in n-body decays
- declearn — Declearn - a python package for private decentralized learning.
- 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.
- deepqmc — Deep-learning quantum Monte Carlo for electrons in real space
- deluca — no summary
- 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.
- dgenerate — Batch image generation and manipulation tool supporting Stable Diffusion and related techniques / algorithms, with support for video and animated image processing.
- dgmrf — Deep Gaussian Markov Random Fields and their extensions
- dibs-lib — DiBS: Differentiable Bayesian Structure Learning
- diffeqzoo — Ordinary differential equation zoo
- differt — Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
- DiffeRT2d — 2D Toolbox for Differentiable Ray Tracing
- diffilqrax — Differentiable iLQR algorithm for dynamical systems
- diffjeom — Differential geometry with jax
- diffqc — Diiferentiable Quantum Simulator
- diffrax — GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
- diffusers — State-of-the-art diffusion in PyTorch and JAX.
- diffusers-unchained — Diffusers
- diffusersv — State-of-the-art diffusion in PyTorch and JAX.
- dimredpy — Dimensionality reduction techniques in python, built on JAX
- dinf — discriminator-based inference for population genetics
- discco — Quantification of membrane and cytoplasmic concentrations based on differentiable simulation of cell cortex images
- discopy — The Python toolkit for computing with string diagrams.
- distrax — Distrax: Probability distributions in JAX.
- dkmri-WIN — Reproducible and efficient diffusion kurtosis imaging in Python.
- dks — A Python library implementing the DKS/TAT neural network transformation method.
- dLux — Differentiable Optical Models as Parameterised Neural Networks in Jax using Zodiax.
- dm-clrs — The CLRS Algorithmic Reasoning Benchmark.
- dm-haiku — Haiku is a library for building neural networks in JAX.
- dm-pix — PIX is an image processing library in JAX, for JAX.
- dMO — A package for learning cutting planes for mixed-integer optimization problems.
- docarray — The data structure for multimodal data
- dopamax — Reinforcement learning in pure JAX.
- dopamine-rl — Dopamine: A framework for flexible Reinforcement Learning research
- drpangloss — Python package to fit interferometric data, accelerated using Jax.
- dsolve — Solver of dynamic equations with forward looking variables
- dsps — Differentiable Stellar Population Synthesis
- dvoryan — Reproducible and efficient diffusion kurtosis imaging in Python.
- dynamax — Dynamic State Space Models in JAX.
- e3nn-jax — Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.
- EasyDeL — An open-source library to make training faster and more optimized in Jax/Flax
- easyjax — Facilitates machine learning development for JAX.
- econpizza — Solve nonlinear perfect foresight models with heterogeneous agents
- efax — Exponential families for JAX
- egnn-jax — E(3) GNN in jax
- einconv — Convolutions as tensor contractions (einsums) for PyTorch
- einshape — DSL-based reshaping library for JAX and other frameworks
- einshard — High-level array sharding API for JAX
- elle — Library of anonymous finite elements with analytic derivatives.
- elrpy — A lightweight package for individual level inference in elections backed by jax.
- emme — Object oriented finite element analysis.
- enformer-dna-diff — This is a package for that combines DeepMind Enformer model with DNA Diffusion project
- equinox — Elegant easy-to-use neural networks in JAX.
- etils — Collection of common python utils
- eulerpi — The eulerian parameter inference (eulerpi) returns a parameter distribution, which is consistent with the observed data by solving the inverse problem directly. In the case of a one-to-one mapping, this is the true underlying distribution.
- evaluators — Various scene understanding and perception evaluation metrics.
- evermore — Differentiable (binned) likelihoods in JAX.