Ecosyste.ms: Issues
An open API service for providing issue and pull request metadata for open source projects.
GitHub / deepmind/optax issues and pull requests
#564 - Fix typo.
Pull Request -
State: open - Opened by copybara-service[bot] over 1 year ago
#563 - Simple optimization problem running faster in Tensor Flow compared to JAX with Optax
Issue -
State: open - Opened by alexandermm over 1 year ago
#562 - The corresponding versions between optax and JAX
Issue -
State: open - Opened by TInaWangxue over 1 year ago
#561 - Update the code to not use soon-to-be-deprecated Chex assertions.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#560 - Update the code to not use soon-to-be-deprecated Chex assertions.
Pull Request -
State: open - Opened by copybara-service[bot] over 1 year ago
#559 - Bump version to 0.1.8.dev
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#558 - Remove deprecated zip_safe field, and potentially fix the deploy workflow
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#557 - Optax 0.1.6 unavailable on PyPi and conda-forge
Issue -
State: closed - Opened by ilyanoskov over 1 year ago
#556 - kl_divergence_with_log_targets is not imported in __init__.py
Issue -
State: open - Opened by h-terao over 1 year ago
#555 - Bump version number now that 0.1.6 is released
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#553 - Casting max_norm such that it can work with types like bfloat16 (currently crashes for bfloat16 updates).
Pull Request -
State: open - Opened by copybara-service[bot] over 1 year ago
- 1 comment
#552 - Wrong documentation formatting
Issue -
State: open - Opened by ayaka14732 over 1 year ago
#551 - Make tree_map_params and masked work together
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#550 - Update the code to not use the deprecated Chex assertions.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#549 - optax: Use specific return-type annotations in loss.py
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#548 - initialize sharderd optimizer state
Issue -
State: open - Opened by gianlucadetommaso over 1 year ago
#547 - Update requirements and release v0.1.6.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#546 - Second-order optimizers in optax
Issue -
State: open - Opened by nalzok over 1 year ago
#545 - Update RTD setup to python 3.11.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#544 - Build Sphinx docs using 3.9.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#543 - Enable pytype checks for >= 3.11
Issue -
State: open - Opened by hbq1 over 1 year ago
#542 - Drop support for 3.8. See https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-14.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#541 - AttributeError: module 'jax.interpreters.xla' has no attribute 'DeviceArray'
Issue -
State: closed - Opened by louieworth over 1 year ago
#540 - Add support for extra args to multi transform
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#539 - Add optimizer code for Mechanic learning rate tuner (https://arxiv.org/pdf/2306.00144.pdf) to Optax.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#537 - Add poly_loss_cross_entropy to public api
Pull Request -
State: open - Opened by acforvs over 1 year ago
#536 - COntinuous COin Betting algorithm
Pull Request -
State: open - Opened by albcab over 1 year ago
- 1 comment
#535 - Fix pylint's warnings in the CI tests.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#533 - jax.interpreters.xla.DeviceArray deprecated in jax 0.4.11
Issue -
State: open - Opened by tisabe over 1 year ago
- 6 comments
#532 - rmsprop implementation in optax and torch
Issue -
State: open - Opened by vwxyzjn over 1 year ago
#531 - Curious if anyone is implementing the Sophia optimizer?
Issue -
State: open - Opened by kszlim over 1 year ago
#530 - Learning rate-free coin betting optimizer
Issue -
State: open - Opened by albcab over 1 year ago
#529 - Added Nadam optimizer
Pull Request -
State: open - Opened by AntoinePlumerault over 1 year ago
#528 - optax.MultiSteps memory consumption
Issue -
State: closed - Opened by albertogaspar over 1 year ago
- 2 comments
#527 - DPSGD won't work correctly with `apply_every`
Issue -
State: open - Opened by long21wt over 1 year ago
#526 - Add empty pyproject.toml to enable PR #513.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#525 - Adding support for Focal Loss?
Issue -
State: open - Opened by paganpasta over 1 year ago
#524 - Resolve pytype errors.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#523 - Release v0.1.5.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#522 - Latest release raises errors with JAX's latest release
Issue -
State: closed - Opened by albcab over 1 year ago
- 1 comment
#519 - Added protobuf versioning to dependencies
Pull Request -
State: closed - Opened by andreped over 1 year ago
- 1 comment
#518 - `optax.inject_hyperparams` no longer works with the latest chex
Issue -
State: closed - Opened by vwxyzjn over 1 year ago
- 4 comments
#517 - Fix pytype failures related to teaching pytype about NumPy scalar types.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#516 - Allow gradient transform parameters to be dynamic
Pull Request -
State: open - Opened by NeilGirdhar over 1 year ago
- 2 comments
#515 - MultiSteps doesn't work with DPSGD
Issue -
State: open - Opened by long21wt over 1 year ago
- 1 comment
#514 - Update pytypes.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#513 - feat: migrate to `pyproject.toml` for building package
Pull Request -
State: closed - Opened by SauravMaheshkar over 1 year ago
- 21 comments
#512 - [JAX] Fix some type annotations in optax that break with an upcoming JAX change.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#511 - Replaces references to jax.numpy.DeviceArray with jax.Array.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
#510 - Make radam docstring point to radam paper.
Pull Request -
State: open - Opened by carlosgmartin over 1 year ago
#509 - """Calculates the squared error for a set of predictions.
Pull Request -
State: closed - Opened by copybara-service[bot] over 1 year ago
- 1 comment
#508 - Help with implementing Adam#Lion optimizer
Issue -
State: open - Opened by buttercutter over 1 year ago
#507 - Add missing optimizers
Issue -
State: open - Opened by carlosgmartin over 1 year ago
#505 - Added reduce_on_plateau
Pull Request -
State: open - Opened by OhadRubin almost 2 years ago
- 2 comments
#504 - Fix JIT for `piecewise_interpolate_schedule`, `cosine_onecycle_schedule`, `linear_onecycle_schedule `
Pull Request -
State: closed - Opened by brentyi almost 2 years ago
#503 - `piecewise_interpolate_schedule` closure leaks tracers under JIT
Issue -
State: closed - Opened by brentyi almost 2 years ago
#502 - chore: move `CONTRIBUTING.md` to `.github/` dir
Pull Request -
State: closed - Opened by SauravMaheshkar almost 2 years ago
- 3 comments
#501 - Add Lion optimizer to Optax.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
- 2 comments
#500 - Add the Lion optimizer, discovered by symbolic program search.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
- 1 comment
#499 - Explicitly export "softmax_cross_entropy_with_integer_labels"
Pull Request -
State: closed - Opened by nasyxx almost 2 years ago
#498 - [JAX] Fix pytype error in Optax if jnp.ndarray == jax.Array.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#497 - Extend the API to support additional arguments.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#496 - Add 3.10 to Github CI.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#495 - fix(README): small typo in README
Pull Request -
State: closed - Opened by jeertmans almost 2 years ago
- 1 comment
#494 - Question: `adamw` with `add_decayed_weights`
Issue -
State: closed - Opened by jenkspt almost 2 years ago
- 1 comment
#493 - [JAX] Replace uses of jax.xla_computation() with jax.jit().lower().
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#492 - Deprecate no longer supported Python 3.7
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#491 - Add kwargs to update
Pull Request -
State: open - Opened by copybara-service[bot] almost 2 years ago
#490 - Run Github CI every day at 03:00.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#489 - Bump ipython from 7.16.3 to 8.10.0 in /requirements
Pull Request -
State: open - Opened by dependabot[bot] almost 2 years ago
Labels: dependencies
#488 - New feature: "tree_map_params"
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#486 - Fix `inject_hyperparams()` for python < 3.10.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#485 - feat(ci/tests): bump `setup-python` version and enable cache
Pull Request -
State: closed - Opened by SauravMaheshkar almost 2 years ago
- 3 comments
#483 - Fix pylint's warnings.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#482 - Cast `labels` to float in `sigmoid_binary_cross_entropy` before the loss is computed. This is meant to guard against unsigned labels (e.g. uint8) being passed in, which results in catastrophic loss values.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#481 - Fix global_norm() signature
Pull Request -
State: closed - Opened by brentyi almost 2 years ago
- 3 comments
#480 - Fix typos in docstring
Pull Request -
State: closed - Opened by pomonam almost 2 years ago
#479 - Documentation Rendering
Issue -
State: open - Opened by lockwo almost 2 years ago
#477 - Freezing parameters: multi_transform + set_to_zero does not work?
Issue -
State: open - Opened by AndreaCossu almost 2 years ago
- 2 comments
#476 - Add exponent to cosine decay schedule and warmup + cosine decay
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#475 - Adds Eve Optimizer
Pull Request -
State: open - Opened by wglao almost 2 years ago
#474 - How to use ema with pmap?
Issue -
State: open - Opened by vighneshbirodkar almost 2 years ago
#473 - fixed NaN issues in `kl_divergence` loss function
Pull Request -
State: closed - Opened by LukasMut almost 2 years ago
- 5 comments
#472 - optax.MultiSteps out of memory
Issue -
State: open - Opened by ein-ich almost 2 years ago
- 3 comments
#470 - Custom optimizer
Issue -
State: closed - Opened by thewizardnet almost 2 years ago
#469 - Add Multi Class Hinge loss
Pull Request -
State: closed - Opened by heytanay almost 2 years ago
#468 - LBFGS support
Issue -
State: open - Opened by jlperla almost 2 years ago
- 2 comments
#467 - Add support for PolyLoss (ICLR 2022)
Pull Request -
State: closed - Opened by acforvs almost 2 years ago
- 8 comments
#466 - Add support for PolyLoss (ICLR 2022)
Issue -
State: closed - Opened by acforvs almost 2 years ago
- 2 comments
#465 - Better tests for utils
Pull Request -
State: closed - Opened by acforvs almost 2 years ago
- 4 comments
#464 - Fix for optimizer name in colab example and api doc, was optimistic_gd but should be optimistic_gradient_descent.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
- 1 comment
#463 - Add support for returning auxiliary values from optimiziers.
Pull Request -
State: closed - Opened by copybara-service[bot] almost 2 years ago
#462 - Change a deprecated chex assert
Pull Request -
State: closed - Opened by copybara-service[bot] about 2 years ago
- 1 comment
#461 - Fix the Yogi optimizer paper year, change link to NeurIPS site
Pull Request -
State: closed - Opened by 8bitmp3 about 2 years ago
#460 - Remove an unused transform state
Pull Request -
State: closed - Opened by copybara-service[bot] about 2 years ago
- 1 comment
#459 - Increase version to 0.1.5.dev.
Pull Request -
State: closed - Opened by copybara-service[bot] about 2 years ago
#458 - Fix arXiv link to Optax Optimistic Gradient Descent optimizer
Pull Request -
State: closed - Opened by 8bitmp3 about 2 years ago
- 1 comment
#457 - [optax] Increase chex version requirement to fix issue #456.
Pull Request -
State: closed - Opened by copybara-service[bot] about 2 years ago
#456 - Can not import optax
Issue -
State: closed - Opened by ankit27kh about 2 years ago
- 3 comments
#455 - [optax] Increase version for new release.
Pull Request -
State: closed - Opened by copybara-service[bot] about 2 years ago