Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• 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-dycore — no summary
• 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.
• 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.
• dm21cm — DM21cm test build
• dMO — A package for learning cutting planes for mixed-integer optimization problems.
• dmpe — Excitation of exciting systems through differentiable predictive excitation
• 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
• 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.
• 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 — An open-source library to make training faster and more optimized in 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
• 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.
• eloy — Pixelated spheres
• 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
• ensemble-kalman-smoother — Ensembling and kalman smoothing for pose estimation
• 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.
• Equimo — Implementation of popular vision models in Jax
• equinox — Elegant easy-to-use neural networks in JAX.
• esch — esch (v.) : to turn matricies into high quality svg (animations)
• esquilax — JAX multi-agent simulation and ML toolset
• 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.
• evo-science — Awesome evo_science created by maycuatroi
• evofr — Tools for evolutionary forecasting.
• evojax — EvoJAX: Hardware-accelerated Neuroevolution.
• evosax — JAX-Based Evolution Strategies
• evox — evox
• ex2mcmc — Local-Global MCMC kernels: the bost of both worlds (NeurIPS 2022)
• exciting-environments — Physical differential equations wrapped into Gymnasium environments
• exojax — Auto-differentiable line-by-line spectral modeling of exoplanets/brown dwarfs using JAX.
• exoplanet-core — The compiled backend for exoplanet
• explainax — JAX-based Model Explanation and Interpretation Library
• exponax — Efficient differentiable PDE solvers in JAX.
• factopy — no summary
• fastax — A Jax based neural network library for research
• fastbrain — Python library of of jax-based representational studies in neuroscience and deep learning
• fastfem — A Python package for solving PDEs with the finite element method and automatic differentiation
• fastreg — Fast sparse regressions
• FContin — Numerical continuation using just the function
• fdtdx — A JAX framework for the finite-difference time-domain (FDTD) method for solving Maxwell's equations with a focus on inverse design of photonic devices.
• fedjax — Federated learning simulation with JAX.
• fedml-afaf — A research and production integrated edge-cloud library for federated/distributed machine learning at anywhere at any scale.
• feedbax — Optimal feedback control + interventions in JAX.
• felafax — felafax
• felupe — Finite Element Analysis
• femr-cuda — Framework for Electronic Medical Records. A python package for building models using EHR data.
• femr-oldcpu — Framework for Electronic Medical Records. A python package for building models using EHR data.
• fenbux — A Simple Statistical Distribution Library In JAX
• fennel-seed — Light-yields for tracks, and cascades
• fibermc — A Jax-based differentiable Monte Carlo estimator with applications to differentiable simulation, computational geometry, and topology optimization.
• fiction — Astronomical image processing with jax
• fiestaEM — Fast inference of electromagnetic signals with JAX
• FiniteDiffX — Finite difference tools in JAX
• FixedPointJAX — Fixed-point iterations for root finding implemented in JAX