Schedule free optimiser implemented in JAX using Optimistix
APACHE-2.0 License
Statistics for this project are still being loaded, please check back later.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Python numerical optimization toolbox
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.sit...
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...
Python 不定期周报
Optax is a gradient processing and optimization library for JAX.
Vision transformer finetuning scripts
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
PageSpeed Insights Tools
Multiple dispatch over abstract array types in JAX.
Helpful scripts for various purposes
Turn SymPy expressions into trainable JAX expressions.
Jätekukko Omakukko API client