google/learned_optimization issues and pull requests

#328 - jax.numpy.clip: update use of deprecated arguments.

Pull Request - State: closed - Opened by copybara-service[bot] 12 days ago

#327 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 2 months ago

#326 - Stackless yashful

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago

#325 - Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#324 - Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change.

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago

#323 - No public description

Pull Request - State: open - Opened by copybara-service[bot] 10 months ago - 1 comment

#322 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#321 - Update README.md

Pull Request - State: open - Opened by Sohl-Dickstein about 1 year ago

#320 - Lopt conv+ln

Pull Request - State: open - Opened by AllanYangZhou about 1 year ago

#319 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#318 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#317 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#316 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#315 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#314 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#313 - Wrong implementation of hyper_v2 mix_layers

Issue - State: open - Opened by mnmjh1215 about 1 year ago

#312 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#311 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#310 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#309 - [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#308 - [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#307 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#306 - Fix various type annotations.

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago

#305 - Ignore incorrect type annotations related to jax dtypes

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#304 - [LSC] Ignore incorrect type annotations related to jax.numpy APIs

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#303 - Issue with the Demo_for_training_a_model_with_a_learned_optimizer.ipynb

Issue - State: open - Opened by SurajSahoo2016 over 1 year ago

#302 - Remove debug and load checkpoints locally

Pull Request - State: closed - Opened by matpalm over 1 year ago

#301 - Ignore incorrect annotations related to chex.PRNGKey

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#300 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#299 - [JAX] Update use of JAX internals in preparation for JAX change.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#298 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#297 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#296 - typo in the tutorial

Issue - State: open - Opened by koszpe over 1 year ago

#295 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#294 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#292 - Remove references to deprecated jax.ShapedArray

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#291 - Fix pytype failures related to teaching pytype about NumPy scalar types.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#290 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#289 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#287 - Suppress some pytype errors related to improved JAX types under pytype.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#286 - [JAX] Fix or ignore type errors from teaching pytype that jnp.ndarray ==

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#285 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#284 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#283 - [JAX] Replace uses of jax.lib.xla_client.Device and jax._src.lib.xla_client.Device with jax.Device.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#282 - pytorch implementation?

Issue - State: open - Opened by chenqq1 almost 2 years ago

#281 - colab demo error

Issue - State: open - Opened by daviddaiiiii almost 2 years ago - 1 comment

#280 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#279 - jnp.sign(mean_rms) is always 1

Issue - State: open - Opened by OhadRubin almost 2 years ago

#278 - Keras integration

Issue - State: open - Opened by rivershah almost 2 years ago

#277 - Change the in_jit check so that it work with both jit and pjit since they are being merge.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#276 - Reshape images to 2D so imshow can show them

Pull Request - State: open - Opened by grofte about 2 years ago

#275 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#274 - Understanding the differences compared to your earlier work & library

Issue - State: open - Opened by ogencoglu about 2 years ago

#273 - Monkey-patch `os.path` to support `gs://` URIs

Pull Request - State: open - Opened by janEbert about 2 years ago

#272 - License of checkpoints

Issue - State: open - Opened by maciejjaskowski about 2 years ago

#271 - Colab link not working

Issue - State: open - Opened by sjmikler about 2 years ago

#270 - Fix or ignore some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago - 1 comment

#269 - Use with Tensorflow JS?

Issue - State: open - Opened by Karric about 2 years ago

#268 - PyTorch port?

Issue - State: open - Opened by derpyplops about 2 years ago - 2 comments

#267 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#266 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#265 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#264 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#263 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#262 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#261 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#260 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#259 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#258 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#257 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#256 - Replace jax.tree* with jax.tree_util.tree* to avoid deprecation warnings.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#255 - Fix flipped variable names. Luckily they were flipped in both places so the code works as-is, but reading and modifying the function is error-prone.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#254 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#253 - last pretrained lopt release

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#252 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#251 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#250 - Added some more entry points which can be run and some prefab evaluation sets

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#249 - Silence type errors generated by new pytype features.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#248 - make pretrained opts work OSS

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#247 - fix oss....

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#246 - less courier imports.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#245 - Don't import courier unless we need to.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#244 - fix opt_to_optax. add oryx as dep.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#243 - Open source some pretrained optimizers.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#242 - Added some docs for how baselines work.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#241 - fix section headings on docs.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#240 - switch rnn mlp lopt to use dictionary.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#239 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#238 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago

#237 - Added tensor summary + more metrics to eval_training

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#236 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago - 1 comment

#235 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago - 1 comment

#234 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago - 1 comment

#233 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago - 1 comment

#232 - bug fix in validation loss targeting -- remove recompute_split.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#231 - Additional hparam sets.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#230 - Use set_params instead of .replace.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#229 - Open source jaxnerf tasks.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#228 - Added new round of normalizers ran out to 300k iterations.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#227 - Internal change

Pull Request - State: open - Opened by copybara-service[bot] over 2 years ago - 1 comment

GitHub / google/learned_optimization issues and pull requests