Commits

Branch selector

 main

User selector

All users

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 29, 2023

• Add `kw_only` option to nn.Module dataclass, and a FLIP doc for it.

  Show description for 790b275
  

  IvyZX

  authored and

  Flax Authors

  committed
  790b275

  Copy full SHA for 790b275

• Make `restore_args_from_target` API use sharding by default.

  Show description for 515393d
  

  IvyZX

  authored and

  Flax Authors

  committed
  515393d

  Copy full SHA for 515393d

• Remove duplicate line in flax imagenet example config

  Show description for 1a605ad
  

  bloops

  authored and

  Flax Authors

  committed
  1a605ad

  Copy full SHA for 1a605ad

• Remove references to deprecated jax.ShapedArray

  Show description for 57a7f26
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  57a7f26

  Copy full SHA for 57a7f26

Commits on Mar 28, 2023

• Updates examples link.

  Show description for 5bfa098
  

  andsteing

  authored and

  Flax Authors

  committed
  5bfa098

  Copy full SHA for 5bfa098

• [Trivial] Uptick the version after new release.

  Show description for e17d1ce
  

  IvyZX

  authored and

  Flax Authors

  committed
  e17d1ce

  Copy full SHA for e17d1ce

• Merge pull request #2987 from IvyZX:push

  Show description for e061c69
  

  Flax Authors

  committed
  e061c69

  Copy full SHA for e061c69

• release 0.6.8

  IvyZX

  committed
  b8c89e8

  Copy full SHA for b8c89e8

Commits on Mar 27, 2023

• Merge pull request #2958 from chiamp:pop_doc_str

  Show description for 8957389
  

  Flax Authors

  committed
  8957389

  Copy full SHA for 8957389

• Merge pull request #2975 from chiamp:frozen_dict_api_doc

  Show description for 396edea
  

  Flax Authors

  committed
  396edea

  Copy full SHA for 396edea

• Merge pull request #2926 from google:missing-references

  Show description for dd8040e
  

  Flax Authors

  committed
  dd8040e

  Copy full SHA for dd8040e

• Copybara import of the project:

  Show description for 2248f44
  

  dependabot[bot]

  authored and

  Flax Authors

  committed
  2248f44

  Copy full SHA for 2248f44

• Copybara import of the project:

  Show description for 02e2ad1
  

  dependabot[bot]

  authored and

  Flax Authors

  committed
  02e2ad1

  Copy full SHA for 02e2ad1

• Merge pull request #2977 from google:dependabot/pip/examples/lm1b/tensorflow-2.11.1

  Show description for f037751
  

  Flax Authors

  committed
  f037751

  Copy full SHA for f037751

• Merge pull request #2979 from google:dependabot/pip/examples/nlp_seq/tensorflow-2.11.1

  Show description for b0a78a0
  

  Flax Authors

  committed
  b0a78a0

  Copy full SHA for b0a78a0

• Merge pull request #2980 from google:dependabot/pip/examples/sst2/tensorflow-2.11.1

  Show description for a7b0242
  

  Flax Authors

  committed
  a7b0242

  Copy full SHA for a7b0242

• Merge pull request #2983 from google:dependabot/pip/examples/imagenet/tensorflow-2.11.1

  Show description for ff55c0d
  

  Flax Authors

  committed
  ff55c0d

  Copy full SHA for ff55c0d

• Merge pull request #2984 from google:dependabot/pip/examples/ppo/tensorflow-2.11.1

  Show description for 9b13d0f
  

  Flax Authors

  committed
  9b13d0f

  Copy full SHA for 9b13d0f

Commits on Mar 25, 2023

• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/ppo

  Show description for b7c5486
  dependabot[bot]

  authored
  b7c5486

  Copy full SHA for b7c5486

• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/imagenet

  Show description for 8af934d
  dependabot[bot]

  authored
  8af934d

  Copy full SHA for 8af934d

• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/nlp_seq

  Show description for 81b432e
  dependabot[bot]

  authored
  81b432e

  Copy full SHA for 81b432e

• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/sst2

  Show description for 742fe76
  dependabot[bot]

  authored
  742fe76

  Copy full SHA for 742fe76

• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/lm1b

  Show description for f939c4d
  dependabot[bot]

  authored
  f939c4d

  Copy full SHA for f939c4d

• Fixed pop utility fn doc string

  chiamp

  committed
  7b1b4d9

  Copy full SHA for 7b1b4d9

Commits on Mar 24, 2023

• Added utility functions to frozen_dict api

  chiamp

  committed
  e1cd5a1

  Copy full SHA for e1cd5a1

• Do not print fields defined with dataclasses.field(repr=False)

  Show description for d384135
  

  Conchylicultor

  authored and

  Flax Authors

  committed
  d384135

  Copy full SHA for d384135

Commits on Mar 22, 2023

• Make Flax checkpointing use `sharding` instead of extracting mesh and pspecs. Also improved some warning messages.

  Show description for 41c9304
  

  IvyZX

  authored and

  Flax Authors

  committed
  41c9304

  Copy full SHA for 41c9304

• Fixes kwargs in convert_to_graphs_tuple_fn.

  Show description for c2da6f9
  

  andsteing

  authored and

  Flax Authors

  committed
  c2da6f9

  Copy full SHA for c2da6f9

• Merge pull request #2770 from cgarciae:add-bidirectional

  Show description for a4a89ae
  

  Flax Authors

  committed
  a4a89ae

  Copy full SHA for a4a89ae

• add Bidirectional module

  cgarciae

  committed
  6412c6c

  Copy full SHA for 6412c6c

Commits on Mar 21, 2023

• Merge pull request #2967 from yechengxi:patch-2

  Show description for 23340bb
  

  Flax Authors

  committed
  23340bb

  Copy full SHA for 23340bb

Commits on Mar 18, 2023

• set axis='batch'

  Show description for 1c6cb6b
  yechengxi

  authored
  1c6cb6b

  Copy full SHA for 1c6cb6b

Commits on Mar 17, 2023

• Roll back the Orbax flag flip.

  Show description for 751f26a
  

  IvyZX

  authored and

  Flax Authors

  committed
  751f26a

  Copy full SHA for 751f26a

Pagination

Previous

Next