google/learned_optimization issues and pull requests

#323 - No public description

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

#322 - No public description

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

#321 - Update README.md

Pull Request - State: open - Opened by Sohl-Dickstein 6 months ago

#320 - Lopt conv+ln

Pull Request - State: open - Opened by AllanYangZhou 6 months ago

#319 - No public description

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

#318 - No public description

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

#317 - No public description

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

#316 - No public description

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

#315 - No public description

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

#314 - No public description

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

#313 - Wrong implementation of hyper_v2 mix_layers

Issue - State: open - Opened by mnmjh1215 7 months ago

#312 - No public description

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

#311 - No public description

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

#310 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] 8 months 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] 9 months 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] 9 months ago

#307 - Silence some pytype errors.

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

#306 - Fix various type annotations.

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

#305 - Ignore incorrect type annotations related to jax dtypes

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

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

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

#303 - Issue with the Demo_for_training_a_model_with_a_learned_optimizer.ipynb

Issue - State: open - Opened by SurajSahoo2016 10 months ago

#302 - Remove debug and load checkpoints locally

Pull Request - State: closed - Opened by matpalm 10 months ago

#301 - Ignore incorrect annotations related to chex.PRNGKey

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

#300 - No public description

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

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

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

#298 - Internal change

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

#297 - Internal change

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

#296 - typo in the tutorial

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

#295 - Internal change

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

#294 - Silence some pytype errors.

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

#292 - Remove references to deprecated jax.ShapedArray

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

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

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

#290 - Internal change

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

#289 - Internal change

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

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

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

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

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

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

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

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

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year 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] over 1 year ago

#282 - pytorch implementation?

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

#281 - colab demo error

Issue - State: open - Opened by daviddaiiiii over 1 year ago - 1 comment

#280 - Internal change

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

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

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

#278 - Keras integration

Issue - State: open - Opened by rivershah over 1 year 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] over 1 year ago

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

Pull Request - State: open - Opened by grofte over 1 year ago

#275 - Silence some pytype errors.

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

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

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

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

Pull Request - State: open - Opened by janEbert over 1 year ago

#272 - License of checkpoints

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

#271 - Colab link not working

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

#270 - Fix or ignore some pytype errors.

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

#269 - Use with Tensorflow JS?

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

#268 - PyTorch port?

Issue - State: open - Opened by derpyplops over 1 year ago - 2 comments

#267 - Internal change

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

#266 - Internal change

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

#265 - Internal change

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

#264 - Internal change

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

#263 - Internal change

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

#262 - Internal change

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

#261 - Internal change

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

#260 - Internal change

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

#259 - Silence some pytype errors.

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

#258 - Silence some pytype errors.

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

#257 - Internal change

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

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

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year 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] almost 2 years ago

#254 - Silence some pytype errors.

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

#253 - last pretrained lopt release

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

#252 - Internal change

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

#251 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] almost 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] almost 2 years ago

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

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

#248 - make pretrained opts work OSS

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

#247 - fix oss....

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

#246 - less courier imports.

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

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

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

#244 - fix opt_to_optax. add oryx as dep.

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

#243 - Open source some pretrained optimizers.

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

#242 - Added some docs for how baselines work.

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

#241 - fix section headings on docs.

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

#240 - switch rnn mlp lopt to use dictionary.

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

#239 - Internal change

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

#238 - Internal change

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

#237 - Added tensor summary + more metrics to eval_training

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

#236 - Internal change

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

#235 - Internal change

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

#234 - Internal change

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

#233 - Internal change

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

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

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

#231 - Additional hparam sets.

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

#230 - Use set_params instead of .replace.

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

#229 - Open source jaxnerf tasks.

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

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

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

#227 - Internal change

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

#226 - Internal change

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

#225 - Internal change

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

#224 - Internal change

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

#223 - Internal change

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

#222 - Internal change

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

GitHub / google/learned_optimization issues and pull requests