-
quax Public
Multiple dispatch over abstract array types in JAX.
-
lineax Public
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
-
optimistix Public
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
-
jaxtyping Public
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
-
diffrax Public
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
-
equinox Public
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
-
wadler_lindig Public
A Wadler--Lindig pretty printer for Python
-
esm2quinox Public
An implementation of ESM2 in Equinox+JAX
-
-
flytekit Public
Forked from flyteorg/flytekitExtensible Python SDK for developing Flyte tasks and workflows. Simple to get started and learn and highly extensible.
Python Apache License 2.0 UpdatedFeb 8, 2025 -
action_update_python_project Public
Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI
-
sympy2jax Public
Turn SymPy expressions into trainable JAX expressions.
-
beartype Public
Forked from beartype/beartypeUnbearably fast near-real-time runtime type-checking in pure Python.
Python MIT License UpdatedOct 1, 2024 -
torchtyping Public
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
-
Deep-Signature-Transforms Public
Code for "Deep Signature Transforms" (NeurIPS 2019)
-
exvoker Public archive
A CLI tool. Extract regexes from stdout (e.g. URLs) and invoke commands on them (e.g. open the webpage).
-
typst_pyimage Public
Typst extension, adding support for generating figures using inline Python code
-
signatory Public
Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)
-
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
pytkdocs_tweaks Public
Some custom tweaks to the results produced by pytkdocs.
-
-
-
torchcde Public
Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
-
sympytorch Public
Turning SymPy expressions into PyTorch modules.
-
levanter Public
Forked from stanford-crfm/levanterLegibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax
Python Apache License 2.0 UpdatedJul 7, 2023 -
haliax Public
Forked from stanford-crfm/haliaxNamed Tensors for Legible Deep Learning in JAX
-
generalised_shapelets Public
Code for "Generalised Interpretable Shapelets for Irregular Time Series"
-
mkposters Public
Make posters from Markdown files.
-
cpython Public
Forked from python/cpythonThe Python programming language
Python Other UpdatedFeb 26, 2023 -
JAXFLUIDS Public
Forked from tumaer/JAXFLUIDSDifferentiable Fluid Dynamics Package
Python Other UpdatedDec 14, 2022