Comments (6)
We now have documentation on read the docs!
https://optax.readthedocs.io/en/latest/
from optax.
Hi, thanks for reaching out.
Yes I can confirm that 'optax' is the supported solution going forward. It is already widely used internally at DeepMind and I encourage you to also use 'optax' rather than 'optix', as the latter will be deprecated soon.
We will be adding more public documentation for 'optax' in the coming weeks.
from optax.
I just wanted to chime in and say that this would be nice to have. I'm working on a project that used to reference some optix functions through intersphinx.. now I don't know where to point it to.
from optax.
Thanks! Will update Elegy then :D
from optax.
switched. nice library!
from optax.
Thanks @mtthss
It's so nice to walk through the API without being distracted by the code itself. (I just saw periodic-update
for the first time, awesome!)
One minor thing.. Some of the sections aren't showing up in the correct hierarchy in the sidebar. For instance, "Apply Updates" is listed as a subsubsection (under its predecessor item).
I changed some ----
sections to ====
in #79
from optax.
Related Issues (20)
- Allow RMSProp use same scaling as adam without momentum + make schedule_free_adamw use rmsprop directly (to spare one memory slot)) HOT 1
- Masking certain parameters for weight decay in adamw HOT 10
- Momentum buffers still created and computed even if beta1 is 0 HOT 4
- Feature Request: Second-Order Optimization Methods in Optax HOT 2
- LBFGS not working for custom classes HOT 1
- Precision bug in `optax.softmax_cross_entropy_with_integer_labels`
- Feature request: GaLore optimizer HOT 2
- Add an assignment problem solver
- Jaxopt vs Optax HOT 3
- LinearSolveTest.test_solve_sparse fails with jax 0.4.26 HOT 1
- How to Cleanly Specify Optimizer, Schedule, & Gradient Clipping with inject_hyperparams HOT 3
- Add ACProp HOT 1
- Intended usage of the Sophia optimiser HOT 7
- Memory leak in optax.radam HOT 2
- Timeline for JaxOpt migration HOT 11
- AttributeError: module 'optax' has no attribute 'lbfgs' HOT 2
- Performance issue: multi_transform and set_to_zero don't prevent computation HOT 1
- Support for CSR format sparse matrix in optimizer? HOT 1
- Loose dependency allows chex version without 'warn_deprecated_function' to be installed HOT 1
- Stochasticsm in Adam HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from optax.