patrick-kidger
Building a JAX ecosystem: Equinox, Diffrax, Lineax, etc. ---- SciML, numerics, neural ODEs ---- Solo traveller, martial artist, scuba diver!
Funding Links: https://github.com/sponsors/patrick-kidger
- Name: Patrick Kidger
- Location: Zürich
- Company: Cradle.bio
- Kind: user
- Followers: 2329
- Following: 2
- Total stars: 10049
- Repositories count: 58
- Created at: 2022-11-13T11:41:52.577Z
- Updated at: 2025-03-25T00:11:09.519Z
- Last synced at: 2025-03-25T00:11:09.519Z
GitHub Sponsors Profile
I work in scientific machine learning: combining deep learning with differential equations.
A few projects I'm proud of:
Differential equation solvers in JAX: Diffrax;
Differential equation solvers in PyTorch: torchsde, torchcde;
Tensor type annotations (inc. shape, dtypes, ...) for JAX/NumPy/PyTorch: jaxtyping;
Neural networks in JAX, elegantly: Equinox.
See also:
My personal website
My Google Scholar page
Please feel free to use my work without sponsorship! A few kind souls asked for this as a way to say thanks :)
- Current Sponsors: 3
- Past Sponsors: 12
- Total Sponsors: 15
- Minimum Sponsorship: $1.00
Featured Works
google-research/torchsde
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Language: Python - Stars: 1624patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Language: Python - Stars: 1574patrick-kidger/torchcde
Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
Language: Python - Stars: 434patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Language: Python - Stars: 2286patrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Language: Python - Stars: 1421Active Sponsors
Past Sponsors
Sponsor Breakdown
- User: 13