Reverse Dependencies of jax

The following projects have a declared dependency on jax:

• piel — Photonic Integrated Electronics: microservices to codesign photonics, electronics, communications, quantum, and more.
• pinnde — A library for solving differential equations with PINNs and DeepONets
• pinnx — Physics-informed.
• piquasso — A simulator for photonic quantum computations.
• piscola — Type Ia Supernova Light-curve fitting code
• planet-harmonica — Light curves for exoplanet transmission mapping.
• plothelp — Simple auto-grid pyplot util.
• pneuma — Pneuma is an LLM-powered data discovery system for tabular data.
• polyatomic-complexes — no summary
• pore2chip — A package that takes 3D images of porous materials and generates representative micromodels
• pormake — Python library for the construction of porous materials using topology and building blocks
• portrait — Compute and visualize observations phase coverage
• POT — Python Optimal Transport Library
• 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: Stateful AI-generated output with a focus on creativity, speed, and structure.
• 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
• pyfixest — Fast high dimensional fixed effect estimation following syntax of the fixest R package.
• pyg3 — An end to end differentiable finite element framework.
• pyHarm-os — A Harmonic Balance code for mechanical vibration systems with nonlinearities.
• 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
• pylcm — Specification, solution, and simulation of finite-horizon dynamic choice models.
• pylexfluent — Librairie outils IA Lexia par Lexfluent
• 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
• PyPIC3D — A 3D Particle-In-Cell simulation package
• 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
• pysolver_view — Visualization of solver convergence evolution
• 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 pytrees in the most intuitive way possible.
• pytreex — no summary
• pytzer — The Pitzer model for chemical activities and equilibria in aqueous solutions in Python
• pywavelet — WDM wavelet transform your time/freq series!
• 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 [39e3b975] - 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.
• qrisp — Qrisp - A high level language for gate-based quantum computing
• 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
• quark-famework — Framework for Quantum Computing Application Benchmarking
• quask — A software framework to speed up the research in quantum machine learning
• quax — Multiple dispatch in JAX via custom interpreters.
• quax-blocks — Building blocks for Quax classes
• quaxed — Pre-quaxed libraries for multiple dispatch over abstract array types in JAX
• qudit-sim — Multi-qudit system simulation and analysis