Cradle.bio
Patrick Kidger
ML+proteins, sciML, numerics, neural ODEs ╱ building 'scipy w/ autodiff+GPU' in JAX: Equinox, Diffrax, Lineax, etc ╱ solo traveller, martial artist, scuba diver
2.9K
65
9.2K
1
Projects
Repositories this builder owns.
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Connected narratives
Related builders
Others building in the same ecosystem.
