schedule_free_optx

Schedule free optimiser implemented in JAX using Optimistix

APACHE-2.0 License

Stars
4
View Code on GitHub View on X
Ecosystems: Python

No README available, please check again later.

Related Projects
PSITools

PageSpeed Insights Tools

12 May 2014 17
django_async

10 Sep 2020 0
equinox

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

29 Jul 2021 1,846
ViT-finetuning-scripts

Vision transformer finetuning scripts

25 Oct 2023 19
tidbits

Helpful scripts for various purposes

01 Feb 2018 7
coraza-demo-stack

29 Aug 2022 5
optax

Optax is a gradient processing and optimization library for JAX.

12 Jun 2020 1,505
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
PythonWeekly

Python 不定期周报

02 May 2020 8
optimistix

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

03 Oct 2023 311
PyOptim

Python numerical optimization toolbox

10 Nov 2016 12
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...

23 Jun 2022 1,131
pytekukko

Jätekukko Omakukko API client

05 Sep 2021 0