Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Sep 19, 2023

• Return sharding whenever the leaf has .sharding attribute, not checking explicitly for jax.Array.

  Show description for 2f2ec1a
  

  IvyZX

  authored and

  Flax Authors

  committed
  2f2ec1a

  Copy full SHA for 2f2ec1a

Commits on Sep 18, 2023

• Merge pull request #3342 from chiamp:flax_basics

  Show description for caba06f
  

  Flax Authors

  committed
  caba06f

  Copy full SHA for caba06f

• Merge pull request #3344 from chiamp:resnet

  Show description for ed2b1cf
  

  Flax Authors

  committed
  ed2b1cf

  Copy full SHA for ed2b1cf

• Update lift to use the new PRNGKey check to suppress deprecation warnings

  Show description for 830d335
  

  Flax Team

  committed
  830d335

  Copy full SHA for 830d335

Commits on Sep 16, 2023

• updated flax_basics

  chiamp

  committed
  7367002

  Copy full SHA for 7367002

Commits on Sep 15, 2023

• changed resnet v1 to v1.5

  chiamp

  committed
  69b3732

  Copy full SHA for 69b3732

• Merge pull request #3343 from chiamp:haiku_upgrade

  Show description for 5d846a5
  

  Flax Authors

  committed
  5d846a5

  Copy full SHA for 5d846a5

• updated haiku guide with new JAX RNG api

  chiamp

  committed
  089f237

  Copy full SHA for 089f237

Commits on Sep 14, 2023

• Migrate flax from using old-style PRNG keys to new-style typed PRNG keys

  Show description for 654ae1a
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  654ae1a

  Copy full SHA for 654ae1a

• Merge pull request #3277 from IvyZX:pen

  Show description for d9f83bf
  

  Flax Authors

  committed
  d9f83bf

  Copy full SHA for d9f83bf

Commits on Sep 13, 2023

• Merge pull request #3334 from google:0.7.5

  Show description for 302a7fa
  

  Flax Authors

  committed
  302a7fa

  Copy full SHA for 302a7fa

• Fix issue with rng check for new-style PRNG keys

  Show description for d6f744a
  

  Jake VanderPlas

  authored and

  Flax Authors

  committed
  d6f744a

  Copy full SHA for d6f744a

• 0.7.5

  cgarciae

  committed
  33b460c

  Copy full SHA for 33b460c

• Merge pull request #3331 from levskaya:version_fix

  Show description for 9c40e03
  

  Flax Authors

  committed
  9c40e03

  Copy full SHA for 9c40e03

• Update minimum python version info.

  Show description for 7ce981b
  levskaya

  committed
  7ce981b

  Copy full SHA for 7ce981b

• Bump version no by 1 post-release.

  Show description for ca3ea06
  

  IvyZX

  authored and

  Flax Authors

  committed
  ca3ea06

  Copy full SHA for ca3ea06

• Merge pull request #3326 from IvyZX:push

  Show description for 5f06f08
  

  Flax Authors

  committed
  5f06f08

  Copy full SHA for 5f06f08

• 0.7.3 release

  IvyZX

  committed
  ad8d22e

  Copy full SHA for ad8d22e

Commits on Sep 12, 2023

• Merge pull request #3314 from jakevdp:typed-prng

  Show description for 9aeb909
  

  Flax Authors

  committed
  9aeb909

  Copy full SHA for 9aeb909

Commits on Sep 11, 2023

• Merge pull request #3281 from google:method-accept-str-submodules

  Show description for a5cf5db
  

  Flax Authors

  committed
  a5cf5db

  Copy full SHA for a5cf5db

• allow method argument to accept submodules

  cgarciae

  committed
  37cbb7d

  Copy full SHA for 37cbb7d

• Internal change

  Show description for 426c0b4
  

  romanngg

  authored and

  Flax Authors

  committed
  426c0b4

  Copy full SHA for 426c0b4

• Internal change

  Show description for a1829ad
  

  romanngg

  authored and

  Flax Authors

  committed
  a1829ad

  Copy full SHA for a1829ad

Commits on Sep 8, 2023

• Merge pull request #3317 from google:fix-main3

  Show description for 23bd52d
  

  Flax Authors

  committed
  23bd52d

  Copy full SHA for 23bd52d

• random change

  cgarciae

  committed
  4f5c143

  Copy full SHA for 4f5c143

• add tf-text to doc requirements

  cgarciae

  committed
  ad8a20f

  Copy full SHA for ad8a20f

• Update handling of typed PRNG keys

  Show description for 9a1496f
  jakevdp

  committed
  9a1496f

  Copy full SHA for 9a1496f

Commits on Sep 7, 2023

• Merge pull request #3308 from anuragarnab:patch-3

  Show description for bec1c80
  

  Flax Authors

  committed
  bec1c80

  Copy full SHA for bec1c80

• [flax] Add module path to nn.module.

  Show description for e3f579a
  

  JXRiver

  authored and

  Flax Authors

  committed
  e3f579a

  Copy full SHA for e3f579a

Commits on Sep 6, 2023

• Minor fix to jax_utils.py

  Show description for 7291174
  anuragarnab

  authored
  7291174

  Copy full SHA for 7291174

Commits on Sep 5, 2023

• Merge pull request #2585 from cgarciae:rnn-flip

  Show description for 82dd366
  

  Flax Authors

  committed
  82dd366

  Copy full SHA for 82dd366

• add rnn flip

  cgarciae

  committed
  9b2f086

  Copy full SHA for 9b2f086

Commits on Sep 4, 2023

• update min typing_extensions version

  Show description for d72ab2c
  

  levskaya

  authored and

  Flax Authors

  committed
  d72ab2c

  Copy full SHA for d72ab2c

Commits on Sep 1, 2023

• Promote dot_general injections from a function to a module.

  Show description for 1127475
  

  Flax Team

  committed
  1127475

  Copy full SHA for 1127475

Commits on Aug 28, 2023

• Allow pass-through implementation of state dict

  Show description for 6fcf8c9
  

  jheek

  authored and

  Flax Authors

  committed
  6fcf8c9

  Copy full SHA for 6fcf8c9