Block or Report
Block or report mthiboust
Contact GitHub support about this user’s behavior. Learn more about reporting abuse.
Report abuseStars
Language: Python
Sort by: Most stars
You like pytorch? You like micrograd? You love tinygrad! ❤️
Simple, elegant, Pythonic functional programming.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
The blessed GitHub Action, for publishing your 📦 distribution files to PyPI: https://github.com/marketplace/actions/pypi-publish
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Multiple dispatch over abstract array types in JAX.