Reverse Dependencies of jaxtyping

The following projects have a declared dependency on jaxtyping:

• apax — Atomistic Learned Potential Package in JAX
• arlatentsde — Amortized Reparametrization for Continuous Time Autoencoders (ARCTA)
• awblib — A bunch of Automatic White-Balancing (AWB) Algorithm implementations
• bayesian-lora — Bayesian LoRA adapters for Language Models
• bayesnf — Scalable spatiotemporal prediction with Bayesian neural fields
• bbhamux — Minimal library to construct Hierarchical Associative Memories
• calcgp — Gaussian Process Regression framework for numerical integration and differentiation
• cfrx — Counterfactual Regret Minimization in Jax
• chemise — Wrapper for training flax models
• chromax — Breeding simulator based on JAX
• compress-em — Particle Picking of Cryo-EM Datasets
• coordinax — Vectors in JAX
• cryojax — Cryo-EM image simulation and analysis powered by JAX
• differt — Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
• diffjpeg_jax — DiffJPEG implemented in JAX
• diffpose — Patient-specific intraoperative 2D/3D registration via differentiable rendering
• diffrax — GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
• dynamax — Dynamic State Space Models in JAX.
• e2e-sae — Repo for training sparse autoencoders end-to-end
• equinox — Elegant easy-to-use neural networks in JAX.
• evermore — Differentiable (binned) likelihoods in JAX.
• feedbax — Optimal feedback control + interventions in JAX.
• fenbux — A Simple Statistical Distribution Library In JAX
• flowjax — Easy to use distributions, bijections and normalizing flows in JAX.
• folx — Forward Laplacian for JAX
• google-vizier — Open Source Vizier: Distributed service framework for blackbox optimization and research.
• google-vizier-dev — Open Source Vizier: Distributed service framework for blackbox optimization and research.
• gpfy — Gaussian process with spherical harmonic features in JAX
• GPJax — Gaussian processes in JAX.
• gpjax-nightly — Didactic Gaussian processes in Jax.
• gradoptorch — Classical gradient based optimization in PyTorch
• grain — Grain: A library for loading and transforming data for neural network training.
• grain-nightly — Grain: A library for loading and transforming data for neural network training.
• graph-pes — Potential Energy Surfaces on Graphs
• graphein — Protein & Interactomic Graph Construction for Machine Learning
• haliax — Named Tensors for Legible Deep Learning in JAX
• holab — Acoustic lens design on GPUs with Automatic Differentiation.
• inseq — Interpretability for Sequence Generation Models 🔍
• janus-sim — A JAX Neural Simulator
• jax_chmc — An implementation of a family of MCMC methods on implicitly defined manifolds
• jax-f16 — Jax F16 dynamics
• jax-healpy — Healpix JAX implementation.
• jax-quantity — Quantities in JAX
• jax-russell — Financial options valuations and auto-diffed risk measures ("greeks" in Jax.
• jaxampler — A JAX-based statistical sampling toolkit
• JaxKern — Kernels in Jax.
• jaxkern-nightly — Kernels in Jax.
• jaxlinop-nightly — A JaxLinOp library.
• jaxonloader — A dataloader, but for JAX/Numpy
• jaxparrow — Computes the inversion of the cyclogeostrophic balance based on a variational formulation approach, using JAX
• jaxrenderer — Jax implementation of rasterizer renderer.
• JaxUtils — Utility functions for JaxGaussianProcesses
• jaxutils-nightly — Utility functions for JaxGaussianProcesses
• jit-env — A Jax interface for Reinforcement Learning environments.
• jo3mnist — no summary
• k-Wave-python — Acoustics toolbox for time domain acoustic and ultrasound simulations in complex and tissue-realistic media.
• kira_llm — That's right, I'm Kira ✍️
• levanter — Scalable Training for Foundation Models with Named Tensors and JAX
• linear-operator — A linear operator implementation, primarily designed for finite-dimensional positive definite operators (i.e. kernel matrices).
• lineax — Linear solvers in JAX and Equinox.
• mamba-lens — TransformerLens port for Mamba
• maze-dataset — no summary
• MCCube — Markov chain cubature via JAX.
• mini-dust3r — Miniature version of dust3r, focused on inference
• moveread-export — Exporting data from annotations of the Moveread Core
• mubelnet — Deep Bayesian unsupervised decoder networks. Use poisson or multinomial belief networks to cluster non-negative count data.
• muutils — A collection of miscellaneous python utilities
• nadl — Nasy's Deep Learning Toolkit
• nerfstudio — All-in-one repository for state-of-the-art NeRFs
• optimistix — Nonlinear optimisation in JAX and Equinox.
• phlash — Bayesian inference of population size history from recombining sequence data.
• praxis — Functionalities such as a layers for building neural networks in Jax.
• proteinworkshop — no summary
• pycollimator — Collimator.ai core simulation engine and API client
• quantumspectra-2024 — generates absorption spectra for donor-acceptor quantum systems
• quax — Multiple dispatch in JAX via custom interpreters.
• refiners — The simplest way to train and run adapters on top of foundation models
• robust-extraction — Automatic preprocessing of chess scoresheets
• sae-vis — Open-source SAE visualizer, based on Anthropic's published visualizer.
• safecheck — Utilities for typechecking, shapechecking and dispatch.
• sax — Autograd and XLA for S-parameters
• segretini-matplottini — Colorful recipes and utilities for Matplolib and Seaborn, to create publication-ready plots
• sentinex — Sentinex: A high level interface aimed towards rapid prototyping and intuitive workflow for JAX.
• signax — Differentiable signature calculations in JAX.
• skshapes — Shape analysis in python
• somap — Self-Organizing Maps in python
• stamox — Accelerate Your Statistical Analysis with JAX.
• statedict2pytree — Converts torch models into PyTrees for Equinox
• t3w — Typed Thin PyTorch Wrapper
• tensorbox — Type annotations and runtime checking for dataclass-like containers of tensors.
• tensorwrap — TensorWrap: A high level TensorFlow wrapper for JAX.
• tf-ctc — CTC with tensorflow, simplified
• tf-ocr — Tools for OCR in tensorflow. Typed with `jaxtyping`
• tf-tools — Useful tools for `tensorflow`, typed with `jaxtyping`
• tfc — Theory of Functional Connections (TFC): A functional interpolation framework with applications in differential equations.
• tinyShakespeareLoader — no summary
• torchmix — Flexible components for transformers 🧩
• trajectory-manifold — Statistically Rigorous ODE Forecasting
• transcoders-slim — A template for python projects in PDM
• transformer-lens — An implementation of transformers tailored for mechanistic interpretability.