Commits

Branch selector

 main

User selector

andsteing

Datepicker

All time

Commit History

Commits on Apr 3, 2023

• Adds colab_tpu_jax_version.ipynb

  Show description for 0dde551
  

  andsteing

  authored and

  Flax Authors

  committed
  0dde551

  Copy full SHA for 0dde551

Commits on Mar 30, 2023

• Adds `config.shuffle_buffer_size` and `config.prefetch`.

  Show description for 71e4432
  

  andsteing

  authored and

  Flax Authors

  committed
  71e4432

  Copy full SHA for 71e4432

Commits on Mar 28, 2023

• Updates examples link.

  Show description for 5bfa098
  

  andsteing

  authored and

  Flax Authors

  committed
  5bfa098

  Copy full SHA for 5bfa098

Commits on Mar 22, 2023

• Fixes kwargs in convert_to_graphs_tuple_fn.

  Show description for c2da6f9
  

  andsteing

  authored and

  Flax Authors

  committed
  c2da6f9

  Copy full SHA for c2da6f9

Commits on Mar 16, 2023

• Adds metrics function that describes main metrics of the experiment.

  Show description for 4efd541
  

  andsteing

  authored and

  Flax Authors

  committed
  4efd541

  Copy full SHA for 4efd541

• Removes wmt/configs/gpu_mixed_precision.py

  Show description for fec4110
  

  andsteing

  authored and

  Flax Authors

  committed
  fec4110

  Copy full SHA for fec4110

Commits on Mar 6, 2023

• Updates hyper parameter sweep config in ogbg_molpcba.

  Show description for cbd1119
  

  andsteing

  authored and

  Flax Authors

  committed
  cbd1119

  Copy full SHA for cbd1119

Commits on Nov 10, 2022

• Catches 502 & other small improvements.

  andsteing

  committed
  a6f67ec

  Copy full SHA for a6f67ec

Commits on Nov 9, 2022

• Updates VM image & uses conda instead of virtualenv.

  andsteing

  committed
  e7757bd

  Copy full SHA for e7757bd

• Updates imagenet,wmt requirements.

  Show description for b7ba28c
  andsteing

  committed
  b7ba28c

  Copy full SHA for b7ba28c

Commits on Aug 17, 2022

• Updates examples/{imagenet,wmt} requirements.

  Show description for f175457
  andsteing

  committed
  f175457

  Copy full SHA for f175457

Commits on Aug 16, 2022

• Removes `flax.optim` & bumps version to 0.6.0

  Show description for cda7a4c
  

  andsteing

  authored and

  Flax Authors

  committed
  cda7a4c

  Copy full SHA for cda7a4c

Commits on Aug 11, 2022

• Removes `flax.optim.dynamic_scale`.

  Show description for 517b763
  

  andsteing

  authored and

  Flax Authors

  committed
  517b763

  Copy full SHA for 517b763

Commits on Aug 5, 2022

• Copies `dynamic_scale.py` from `optim/` to `training/`.

  Show description for 868d359
  

  andsteing

  authored and

  Flax Authors

  committed
  868d359

  Copy full SHA for 868d359

Commits on Aug 2, 2022

• Internal change

  Show description for 48dfc05
  

  andsteing

  authored and

  Flax Authors

  committed
  48dfc05

  Copy full SHA for 48dfc05

Commits on Jul 14, 2022

• Adds Reference to big_vision, t5x to "Repositories Using Flax".

  Show description for 9eddf86
  

  andsteing

  authored and

  Flax Authors

  committed
  9eddf86

  Copy full SHA for 9eddf86

Commits on Jun 27, 2022

• Makes `convert_pre_linen()` return same type as arg.

  Show description for 2e04288
  

  andsteing

  authored and

  Flax Authors

  committed
  2e04288

  Copy full SHA for 2e04288

Commits on Jun 15, 2022

• Reply to suggestions from andsteing.

  Show description for 3a3df8d
  andsteing

  committed
  3a3df8d

  Copy full SHA for 3a3df8d

Commits on May 31, 2022

• Updates dynamic_scale in training state.

  Show description for 9d4505b
  

  andsteing

  authored and

  Flax Authors

  committed
  9d4505b

  Copy full SHA for 9d4505b

Commits on May 30, 2022

• Fixes small typos in design_notes/module_lifecycle.rst

  Show description for b3236ce
  

  andsteing

  authored and

  Flax Authors

  committed
  b3236ce

  Copy full SHA for b3236ce

Commits on May 19, 2022

• Replaces last `vs_p` with `variables`.

  andsteing

  committed
  d0822db

  Copy full SHA for d0822db

Commits on May 16, 2022

• Adds `static_return` arg and updates rst/Colab.

  Show description for f5dec70
  andsteing

  committed
  f5dec70

  Copy full SHA for f5dec70

Commits on May 12, 2022

• Adds reference to DarshanDeshpande/jax-models

  Show description for 09802df
  andsteing

  authored
  09802df

  Copy full SHA for 09802df

Commits on May 10, 2022

• Replaced `vs[_p]` with `variables`.

  andsteing

  committed
  63e371a

  Copy full SHA for 63e371a

• Adds "full eval" HOWTO.

  andsteing

  committed
  5a34691

  Copy full SHA for 5a34691

Commits on Apr 27, 2022

• Makes GPU info shorter (-L flag).

  andsteing

  committed
  5a7a345

  Copy full SHA for 5a7a345

Commits on Apr 26, 2022

• Makes annotated_mnist use Optax's xent loss.

  andsteing

  committed
  e298d6c

  Copy full SHA for e298d6c

Commits on Apr 13, 2022

• Updates `clu.metrics.CollectingMetric` to always use `np.array()`.

  Show description for 2387439
  

  andsteing

  authored and

  Flax Authors

  committed
  2387439

  Copy full SHA for 2387439

Commits on Mar 29, 2022

• Clarifies `optim.Adam(weight_decay)` parameter.

  Show description for b4e9b12
  

  andsteing

  authored and

  Flax Authors

  committed
  b4e9b12

  Copy full SHA for b4e9b12

Commits on Mar 23, 2022

• Bumps flax/jax versions in examples/imagenet.

  andsteing

  committed
  2e315cf

  Copy full SHA for 2e315cf

• Bumps version to 0.4.1

  andsteing

  committed
  0377fd4

  Copy full SHA for 0377fd4

Commits on Mar 8, 2022

• Removes dynamic_scale from flax.optim.

  Show description for 55829ad
  

  andsteing

  authored and

  Flax Authors

  committed
  55829ad

  Copy full SHA for 55829ad

Commits on Mar 4, 2022

• Fixes indentation.

  Show description for 55041e9
  

  andsteing

  authored and

  Flax Authors

  committed
  55041e9

  Copy full SHA for 55041e9

Commits on Feb 1, 2022

• Adds deprecation warnings to `flax.optim`.

  andsteing

  committed
  5ba0bc0

  Copy full SHA for 5ba0bc0

• Moves main points from #231 to examples/README

  andsteing

  committed
  bdeb8ad

  Copy full SHA for bdeb8ad

Pagination

Previous

Next