Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• 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
• damast — Package to improve the development of transparent, replicable data processing pipelines
• 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 — Massively Parallel Hyperparameter Optimization for Machine Learning
• 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
• differt — Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
• DiffeRT2d — 2D Toolbox for Differentiable Ray Tracing
• DIFFICE-jax — A differentiable neural-network solver for data assimilation of ice shelves written in JAX
• diffilqrax — Differentiable iLQR algorithm for dynamical systems
• diffjeom — Differential geometry with jax
• diffopt — Parallelization and optimization of Jax models
• 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-qaic — 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
• dinosaur — Differentiable Dynamics for Global Atmospheric Modeling 🦖
• dinosaur-dycore — no summary
• dire-jax — A JAX-based Dimension Reducer
• 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.
• distreqx — Lightweight JAX library of probability distributions and bijectors.
• distributed-kron — An implementation of PSGD Kron optimizer in JAX/optax for large scale distributed training.
• 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.
• dm-robotics-transformations — A library for Rigid Body Transformations
• dm21cm — DM21cm test build
• dMO — A package for learning cutting planes for mixed-integer optimization problems.
• dmpe — Excitation of exciting systems through differentiable model predictive excitation
• dnallm — A toolkit for fine-tuning and inference with DNA Language Models
• dnax — DNA models implemented in JAX
• docarray — The data structure for multimodal data
• dolphin — Workflows for generating surface displacement maps using InSAR
• dopamax — Reinforcement learning in pure JAX.
• dopamine-rl — Dopamine: A framework for flexible Reinforcement Learning research
• driftbench — Python tool to analyse process drifts
• drjax — DrJAX - Scalable and Differentiable MapReduce Primitives in JAX.
• drpangloss — Python package to fit interferometric data, accelerated using Jax.
• drvi-py — Disentangled Generative Representation of Single Cell Omics
• dsbundle — Streamline your data science setup with dsbundle in one effortless install.
• 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.
• dynamaxsys — dynamical systems with jax
• dynamiqs — High-performance quantum systems simulation with JAX (GPU-accelerated & differentiable solvers).
• e3md — Library for machine learning on physical tensors
• e3nn-jax — Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.
• EasyDeL — Accelerate, Optimize performance with streamlined training and serving options with JAX.
• easyjax — Facilitates machine learning development for JAX.
• ebcc — Coupled cluster calculations on electron-boson systems
• econpizza — Solve nonlinear perfect foresight models with heterogeneous agents
• editax — Automatically created RL Environment Editors
• efax — Exponential families for JAX
• eformer — (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX
• egnn-jax — E(3) GNN in jax
• Eikonax — Differentiable Solver for the Anisotropic Eikonal Equation on Triangulated Meshes
• 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 — A namespace for finite element libraries targeted at research into neural network architectures.
• elrpy — A lightweight package for individual level inference in elections backed by jax.
• eminus — Pythonic electronic structure theory.
• emme — Object oriented finite element analysis.
• enformer-dna-diff — This is a package for that combines DeepMind Enformer model with DNA Diffusion project
• ensemble-kalman-smoother — Ensembling and kalman smoothing for pose estimation
• enzax — Differentiable models of enzyme-catalysed reaction networks
• eqr — A python package to perform quantile regression with more meaningful epistemic uncertainty estimates through training set augmentation using synthetic data. A blend of discriminative and generative models.