Owner

patrick-kidger

7 indexed repositories · View on GitHub

• equinox

  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

  Python · 2973 stars

• diffrax

  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  Python · 2105 stars

• jaxtyping

  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

  Python · 1873 stars

• optimistix

  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

  Python · 623 stars

• typst_pyimage

  Typst extension, adding support for generating figures using inline Python code

  Python · 119 stars

• typsy

  Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.

  Typst · 30 stars

• patstdlib

  Some standard pieces I find useful in Typst: fonts, subfigures, algorithms, labeled enums, etc.

  Typst · 13 stars