-
https://github.com/google-deepmind/optax/blob/main/examples/lbfgs.ipynb
-
Hi,
I am having trouble implementing optax LBFGS with equinox types. I am trying to run a linear regression model using this notebook https://github.com/ubcecon/ECON622/blob/master/lectures/lectures/…
-
### Description
Last week a [discussion](https://x.com/A_K_Nain/status/1846222510062285035) took place on Twitter where many people noticed that the performance of JAX on GPUs is still subpar compa…
-
## Description
Currently we have been unable to reproduce the schedule free adamw results with JAX.
There seem to be differences between the optax implementation of schedule-free adamw and the pyto…
-
I would like to know what is the best way to freeze parameters in a model using `nnx` and `optax` (https://flax.readthedocs.io/en/latest/guides/training_techniques/transfer_learning.html#optax-multi-t…
-
### Description
I want to save the executable file generated by jax.jit in the main process and execute this executable file in another process. However, I have found that the performance of the exec…
-
Provide as much information as possible. At least, this should include a description of your issue and steps to reproduce the problem. If possible also provide a summary of what steps or workarounds y…
-
Hi!
Thank you so much for publishing the pipeline but I am having some errors getting the pipeline working properly.
Starting with the igraph version error, adding in louvain and cairocffi as …
-
Hi, are there any plans to port [OSQP](https://jaxopt.github.io/stable/_autosummary/jaxopt.OSQP.html#jaxopt.OSQP), or other ADMM solvers from JaxOpt to Optax soon?
Thanks :)
-
It seems like the first call to `step` of the `GradientDescent` optimizer doesn't perform the step operation. I didn't check if this occurs for other optimizers or do other digging, but can do so if t…