Reverse Dependencies of jax
The following projects have a declared dependency on jax:
- paltax — Strong lensing package using jax
- pandas-toolkit — A collection of pandas accessors to help with common machine/deep learning related functionality.
- panpipes — Panpipes - multimodal single cell pipelines
- par-segmentation — Cell cortex segmentation and quantification in C. elegans PAR protein images
- paramdiffusion — A package for deriving continuous score-based data solutions
- patato — A Python Toolkit for the analysis of photoacoustic tomography data.
- pax3 — A stateful pytree library for training neural networks.
- paxlib — JAX-like API on top of PyTorch
- paxml — Framework to configure and run machine learning experiments on top of Jax.
- pcgym — Reinforcement learning suite of process control problems.
- penzai — Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
- perceiver-jax — Perceiver - JAX
- pfb-imaging — Radio interferometric imaging suite base on the pre-conditioned forward-backward algorithm
- pgmax — Loopy belief propagation for factor graphs on discrete variables in JAX
- pgx — GPU/TPU-accelerated parallel game simulators for reinforcement learning (RL)
- phlash — Bayesian inference of population size history from recombining sequence data.
- pidgen2 — Tools for particle identification (PID) correction for LHCb detector
- piel — Photonic Integrated Electronics: microservices to codesign photonics, electronics, communications, quantum, and more.
- piquasso — no summary
- planet-harmonica — Light curves for exoplanet transmission mapping.
- plothelp — Simple auto-grid pyplot util.
- poincare — Simulation of dynamical systems.
- pormake — Python library for the construction of porous materials using topology and building blocks
- portrait — Compute and visualize observations phase coverage
- 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
- 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
- psfmodels — Scalar and vectorial models of the microscope point spread function (PSF).
- 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
- PyColorIT — no summary
- 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 — The generalized, nodalized HGF for predictive coding.
- pykronecker — Tools for performing efficient Kronecker product-based matrix operations
- 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)
- pyRDDLGym-jax — pyRDDLGym-jax: JAX compilation of RDDL description files, and a differentiable planner in JAX.
- pyscfad — PySCF with autodiff
- pysersic — A Tool for fitting sersic profiles in python
- pySPFM — A python package for Paradigm Free Mapping (3dPFM and 3dMEPFM).
- 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
- 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 [8c65a89] - 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.
- qiskit-dynamics — Qiskit ODE solver
- 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
- quantumspectra-2024 — generates absorption spectra for donor-acceptor quantum systems
- quask — A software framework to speed up the research in quantum machine learning
- quax — Multiple dispatch in JAX via custom interpreters.
- quaxed — Array-API JAX compatibility
- 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
- 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
- rax — Learning-to-Rank using JAX.
- 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
- rejectax — rejectax
- renge — Infer gene regulatory networks from time-series single-cell CRISPR data.
- renn — Research tools for Reverse Engineering Neural Networks (RENN).