Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• powerpax — Small collection of useful utilities for JAX
• praxis — Functionalities such as a layers for building neural networks in Jax.
• prfr — Probabilitic random forest regression algorithm
• probabilistic-reconciliation — Probabilistic reconciliation of time series forecasts
• probdiffeq — Probabilistic numerical solvers for differential equations
• probfindiff — Probabilistic numerical finite differences
• probjax — Jax library for probabilistic computations
• probnum — Probabilistic Numerics in Python.
• progen-transformer — Protein Generation (ProGen)
• protes — Method PROTES (PRobabilistic Optimizer with TEnsor Sampling) for derivative-free optimization of the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) format
• proust — Online time series analysis
• PSE — Proxy Structuring Engine: Ensures AI-generated output adheres to predefined schemas without compromising creativity, speed, or context.
• psfmodels — Scalar and vectorial models of the microscope point spread function (PSF).
• psgd-jax — An implementation of PSGD optimizer in JAX.
• ptyrodactyl — Ptychography through differentiable programming in JAX
• pyanno4rt — A package for Python-based advanced numerical nonlinear optimization in radiotherapy.
• pybamm — Python Battery Mathematical Modelling
• pychastic — Solvers for stochastic differential equations
• pycollimator — Collimator.ai core simulation engine and API client
• pydiffuser — A simulation framework for nonequilibrium statistical physics
• pyegsnrc — no summary
• pyei — no summary
• pyemits — python package for easy manipulation on time series data for quick insight
• pyg3 — An end to end differentiable finite element framework.
• pyhf — pure-Python HistFactory implementation with tensors and autodiff
• pyhgf — Dynamic neural networks for predictive coding
• pykronecker — Tools for performing efficient Kronecker product-based matrix operations
• pyloric-network-simulator — Pure-Python, JAX-accelerate implementation of the pyloric circuit model described by Prinz et al (Nat. Neurosci., 2004)
• pymanopt — Toolbox for optimization on Riemannian manifolds with support for automatic differentiation
• pymbar — Python implementation of the multistate Bennett acceptance ratio (MBAR) method
• pymob — Modelling platform for Python
• pymoto — A modular approach for topology optimization
• pymultipleis — A library for fitting a sequence of electrochemical impedance spectra (JAX version).
• pynajax — A JAX backend for pynapple.
• pyneurons — Composable neurons in JAX
• pynsy — A Python analysis framework.
• pypesto — python-based Parameter EStimation TOolbox
• pyqg-jax — Quasigeostrophic model in JAX (port of PyQG)
• pyqtorch — An efficient, large-scale emulator designed for quantum machine learning, seamlessly integrated with a PyTorch backend. Please refer to https://pyqtorch.readthedocs.io/en/latest/ for setup and usage info, along with the full documentation.
• pyRDDLGym-jax — pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.
• pyrometheus — Code generation for combustion thermochemistry, based on Cantera.
• pyrovelocity — A multivariate RNA Velocity model to estimate future cell states with uncertainty using probabilistic modeling with pyro.
• pyscfad — PySCF with autodiff
• pysersic — A Tool for fitting sersic profiles in python
• pySPFM — A python package for Paradigm Free Mapping (3dPFM and 3dMEPFM).
• pysr — Simple and efficient symbolic regression
• pytensor — Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
• python-auditory-toolbox — Several simple auditory models in JAX, Numpy and Torch
• pytorch2jax — Convert PyTorch models to Jax functions and Flax models
• pytree2safetensors — A simple package to save and load JAX PyTrees to and from Safetensors
• pytreeclass — Visualize, create, and operate on JAX PyTree in the most intuitive way possible.
• pytreex — no summary
• pytzer — The Pitzer model for chemical activities and equilibria in aqueous solutions in Python
• qadence — Pasqal interface for circuit-based quantum computing SDKs
• qarray — Qarray, a GPU accelerated quantum dot array simulator, leveraging parallelised Rust and JAX XLA acceleration
• qat-core — Module qat-core [20c0524a] - Compiled by Eviden
• qax — A JAX transform for writing things which pretend to be tensors
• qdax — A Python Library for Quality-Diversity and NeuroEvolution
• qdeep — Simple Deep Q Learning framework.
• QFI-Opt — This repository contains codes for optimizing the quantum fisher information (QFI) of quantum systems.
• qhdopt — A software package for nonconvex optimization with quantum devices.
• qiskit-addon-aqc-tensor — Approximate quantum compilation with tensor networks
• qiskit-addon-sqd — Classically postprocess noisy quantum samples to yield more accurate energy estimations
• qiskit-dynamics — Qiskit ODE solver
• qore — no summary
• qpax — Differentiable QP solver in JAX.
• qsarKit — A Python package that offers robust predictive modeling using QSAR for evaluating the transfer of environmental contaminants in breast milk. It integrates multiple predictive models, provides synthetic data generation via GANs, and is tailored for researchers and health professionals.
• quadax — Numerical quadrature with JAX
• quafel — no summary
• quantizers — no summary
• quantum-simulation-recipe — Recipe for quantum simulation
• quantumspectra-2024 — generates absorption spectra for donor-acceptor quantum systems
• QuaPy — QuaPy: a framework for Quantification in Python
• quask — A software framework to speed up the research in quantum machine learning
• quax — Multiple dispatch in JAX via custom interpreters.
• quaxed — Pre-quaxed libraries for multiple dispatch over abstract array types in JAX
• qudit-sim — Multi-qudit system simulation and analysis
• queuinx — Queuinx: A library for performance evaluation in Jax
• quipcell — Fine-scale cellular deconvolution based on Generalized Cross Entropy
• qujax — Simulating quantum circuits with JAX
• QUnfold — A quantum-based approach to the distribution unfolding problem
• qutip-jax — JAX backend for QuTiP
• qutip-qoc — The QuTiP quantum optimal control package
• ramsey — Probabilistic deep learning using JAX
• rationalnets — JAX/Flax implementation of rational neural nets
• raw-speech-classification — Trains CNN classifiers from raw speech using Keras and tests them.
• rax — Learning-to-Rank using JAX.
• reax — REAX: A simple training framework for JAX-based projects
• recsim-ng — RecSim NG: Toward Principled Uncertainty Modeling for Recommender Ecosystems
• redco — no summary
• ref-gsd — Reference implementation of generalised score distribution in python
• refx — no summary
• regmod — General regression models
• reinforced-lib — Reinforcement learning library
• rejax — Vectorizable RL algorithms in pure JAX
• rejectax — rejectax
• renge — Infer gene regulatory networks from time-series single-cell CRISPR data.
• renn — Research tools for Reverse Engineering Neural Networks (RENN).
• rex-lib — Robotic Environments with jaX (REX)
• rheedium — Differentiable RHEED simulations in JAX