issues
search
google
/
jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
https://jaxopt.github.io
Apache License 2.0
939
stars
66
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Implement our own version of SGD (+momentum), Adam
#450
fabianp
opened
1 year ago
0
fix cond in zoom linesearch for non-jittable case, moved if_else_cond…
#449
vroulet
closed
1 year ago
0
Hager-Zhang linesearch does not work with non-jittable functions
#448
mblondel
opened
1 year ago
0
Re-enable broken tests for GP bandit.
#447
copybara-service[bot]
closed
1 year ago
0
Re-implement `tree_util.tree_inf_norm` so that it works for pytrees that include empty arrays.
#446
copybara-service[bot]
closed
1 year ago
0
Optimization of complex parameters
#445
geggo
opened
1 year ago
2
How to prevent jitting in jaxopt solver?
#444
itk22
closed
1 year ago
3
AndersonAcceleration mismatch between code and documentation
#443
sophie-xhonneux
closed
1 year ago
1
Revamped zoom_linesearch, wrapped linesearch utilitaries
#442
vroulet
closed
1 year ago
0
Implementing a custom optimizer
#441
itk22
closed
1 year ago
5
Drop Python 3.7 support.
#440
mblondel
closed
1 year ago
0
LBFGSB returns parameters outside of bounds
#439
javier-garcia-tilburg
closed
1 year ago
11
Convert zoom linesearch as iterative linesearch solver
#438
vroulet
closed
1 year ago
5
Make Hager-Zhang linesearch robust to NaN and Inf values.
#437
copybara-service[bot]
closed
1 year ago
0
`NameError: unbound axis name` when running example in documentation
#436
paupereira
opened
1 year ago
2
Fix typos in perturbed docs
#435
gdalle
closed
1 year ago
3
Release v0.7.
#434
mblondel
closed
1 year ago
0
Getting the stacked outputs of FixedPointIteration.run()
#433
sophie-xhonneux
opened
1 year ago
1
Implement a `split_rngs` mechanism, like in `flax.linen.scan`
#432
zaccharieramzi
opened
1 year ago
2
Native optimization methods failing on sharp function which scipy wrapper succeeds on
#430
ry-dgel
opened
1 year ago
0
Allow injecting grad_fun and value_and_grad_fun directly.
#429
copybara-service[bot]
closed
1 year ago
0
Callback function requires 2 arguments when using method='trust-constr'
#428
martin-millon
opened
1 year ago
0
ptxas version issue
#427
zohimchandani
closed
1 year ago
2
Add Broyden in the DEQ example
#426
zaccharieramzi
opened
1 year ago
1
Fixed the iMAML tutorial
#425
zaccharieramzi
closed
1 year ago
8
Use `jaxopt.LBFGS` on `Haiku.Module` Parameters?
#424
Xemin0
closed
1 year ago
9
Terminate Hager-Zhang bracketing when a NaN is encountered, to avoid an infinite loop. Add a `failed` bool to backtracking and H-Z linesearch.
#423
copybara-service[bot]
closed
1 year ago
0
jaxopt.Bisection not jittable, vmappable
#422
deasmhumhna
closed
1 year ago
2
Fix prox_lasso, prox_elastic_net, to handle pytrees
#420
vroulet
closed
1 year ago
0
Twice differentiability of OSQP
#419
nomoriel
opened
1 year ago
3
Added option to use control variate in Jacobian of perturbed_argmax.
#418
LawrenceMMStewart
closed
1 year ago
1
Tiny issues w.r.t data sampling in few-shot iMAML example
#417
zaccharieramzi
closed
1 year ago
3
Implemented (Good) Broyden
#416
zaccharieramzi
closed
1 year ago
8
Constraining output in ScipyRootFinding.
#415
sagarwal-atg
opened
1 year ago
1
deep_equilibrium_model.ipynb code still working?
#414
nihcompmed
closed
1 year ago
0
Added Option to Reduce Variance of Jacobian of Perturbed Argmax
#413
LawrenceMMStewart
closed
1 year ago
4
avoid closing over dynamic jax tracers in the bisection solver
#412
froystig
closed
1 year ago
0
enable `pip` caching in CI + add `pyproject.toml`
#411
SauravMaheshkar
closed
1 year ago
4
Add L-BFGS-B.
#410
copybara-service[bot]
closed
1 year ago
0
jaxopt returns the initial values as the final solution for binary logit model
#409
PepSalehi
closed
1 year ago
3
Add pinball loss for quantile regression
#408
georgedahl
opened
1 year ago
0
Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.
#407
copybara-service[bot]
closed
1 year ago
0
Implicit Differentiation with jaxopt.ScipyMinimize
#406
Wenlin-Chen
closed
1 year ago
1
Added case of functions with outputs in any dimension in perturbations.
#405
copybara-service[bot]
closed
1 year ago
0
Errors with `ScipyBoundedMinimize`
#403
Smit-create
opened
1 year ago
2
Misc improvements in resnet_flax example.
#401
fabianp
closed
1 year ago
2
Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.
#400
copybara-service[bot]
closed
1 year ago
0
Newton's method?
#399
ahwillia
opened
1 year ago
2
Add isotonic module.
#398
mblondel
closed
1 year ago
0
Update requirements.
#397
mblondel
closed
1 year ago
0
Previous
Next