Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 5, 2024

• Update references to JAX's GitHub repo

  Show description for e84959c
  

  jakeharmon8

  authored and

  fedjax authors

  committed
  e84959c

Commits on Aug 21, 2024

• [pmap no rank reduce cleanup]: Update fedjax to use indexing

  Show description for af84c32
  

  pschuh

  authored and

  fedjax authors

  committed
  af84c32

Commits on Jul 25, 2024

• [numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

  Show description for 64cbc49
  

  hawkinsp

  authored and

  fedjax authors

  committed
  64cbc49

Commits on May 15, 2024

• Drop support for Python 3.8 and 3.9

  Show description for ce348b0
  

  kho

  authored and

  fedjax authors

  committed
  ce348b0

• Replace deprecated jax.tree_* functions with jax.tree.*

  Show description for 1069448
  

  Jake VanderPlas

  authored and

  fedjax authors

  committed
  1069448

Commits on Oct 17, 2023

• Suppress some pytype errors.

  Show description for f4c1f00
  

  rxsang

  authored and

  fedjax authors

  committed
  f4c1f00

Commits on Sep 26, 2023

• [LSC] Ignore incorrect type annotations related to jax.numpy APIs

  Show description for 514981f
  

  Jake VanderPlas

  authored and

  fedjax authors

  committed
  514981f

Commits on Aug 23, 2023

• Replace uses of jnp.array in types with jnp.ndarray.

  Show description for ac506be
  

  hawkinsp

  authored and

  fedjax authors

  committed
  ac506be

Commits on Aug 10, 2023

• [JAX] Replace jax.numpy.NINF with -jax.numpy.inf.

  Show description for 3ae290b
  

  hawkinsp

  authored and

  fedjax authors

  committed
  3ae290b

• Change jax.numpy.NINF reference (newly deprecated) to -jax.numpy.inf

  Show description for 813fd0d
  

  FedJAX Team

  committed
  813fd0d

Commits on Jul 10, 2023

• Cut release v0.0.17

  Show description for a478473
  

  kho

  authored and

  fedjax authors

  committed
  a478473

Commits on Apr 10, 2023

• Merge pull request #297 from Lando-L:adaptive_personalized_federated_learning

  Show description for ac4e0bc
  

  fedjax authors

  committed
  ac4e0bc

• Add adaptive personalized federated learning

  

  Lando-L

  authored and

  Lando-L

  committed
  becd9c9

Commits on Mar 21, 2023

• Suppress some pytype errors related to improved JAX types under pytype.

  Show description for 25a765e
  

  hawkinsp

  authored and

  fedjax authors

  committed
  25a765e

Commits on Mar 15, 2023

• [JAX] Update some type annotations to fix some pytype errors.

  Show description for 0fca45d
  

  hawkinsp

  authored and

  fedjax authors

  committed
  0fca45d

Commits on Mar 14, 2023

• Adding an abstraction DefaultWordTokenizer and using it in StackoverflowTokenizer.

  Show description for 4009d5b
  

  stheertha

  authored and

  fedjax authors

  committed
  4009d5b

Commits on Mar 10, 2023

• Silence some pytype errors.

  Show description for 7640d9f
  

  rchen152

  authored and

  fedjax authors

  committed
  7640d9f

Commits on Mar 2, 2023

• Reduce device memory usage in for_each_client & tree_{sum,mean} by donating intermediate values

  Show description for fa8682b
  

  kho

  authored and

  fedjax authors

  committed
  fa8682b

• Drop Python 3.7 support

  Show description for f884fbe
  

  kho

  authored and

  fedjax authors

  committed
  f884fbe

Commits on Mar 1, 2023

• Add example of stateful federated averaging where a counter tracking the total number of training steps taken for each client is maintained and updated across rounds

  Show description for c4f314b
  

  jaehunro

  authored and

  fedjax authors

  committed
  c4f314b

Commits on Feb 28, 2023

• Fix typo in for_each_client docstring

  Show description for dd7238e
  

  jaehunro

  authored and

  fedjax authors

  committed
  dd7238e

Commits on Feb 21, 2023

• Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

  Show description for 689e093
  

  hawkinsp

  authored and

  fedjax authors

  committed
  689e093

Commits on Jan 13, 2023

• Cut version 0.0.16 for release

  Show description for b6ed0fc
  

  jaehunro

  authored and

  fedjax authors

  committed
  b6ed0fc

Commits on Dec 15, 2022

• [NumPy] Remove references to deprecated NumPy type aliases.

  Show description for f57e045
  

  hawkinsp

  authored and

  fedjax authors

  committed
  f57e045

Commits on Nov 22, 2022

• Modify cross entropy loss to allow dense targets.

  Show description for b47ce2c
  

  stheertha

  authored and

  fedjax authors

  committed
  b47ce2c

Commits on Nov 5, 2022

• Adafactor optimizer

  Show description for aeab691
  

  stheertha

  authored and

  fedjax authors

  committed
  aeab691

Commits on Oct 13, 2022

• v0.0.15

  Show description for 05e4849
  

  kho

  authored and

  fedjax authors

  committed
  05e4849

• Require sphinx>=4.3.0 so doc-building works under Python 3.10

  Show description for 7ba7c35
  

  kho

  authored and

  fedjax authors

  committed
  7ba7c35

• Update various Python packaging configs

  Show description for 8693a58
  

  kho

  authored and

  fedjax authors

  committed
  8693a58

Commits on Sep 30, 2022

• Match both jax.Array and DeviceArray. jax.Array is the future jax type that will replace DA, SDA and GDA.

  Show description for 2ee3c1d
  

  yashk2810

  authored and

  fedjax authors

  committed
  2ee3c1d

Commits on Sep 29, 2022

• Remove flax optim test now that flax.optim is deprecated in favor of optax google/flax#2273

  Show description for 180530a
  

  jaehunro

  authored and

  fedjax authors

  committed
  180530a

• Replace jax.xla.DeviceArray private type with the new public type jax.Array.

  Show description for 937be98
  

  yashk2810

  authored and

  fedjax authors

  committed
  937be98

Commits on Aug 2, 2022

• Disable pytype import error for old stax import path

  Show description for 362a355
  

  Jake VanderPlas

  authored and

  fedjax authors

  committed
  362a355

Commits on Jul 28, 2022

• Makes flax.optim import future proof.

  Show description for 4326b85
  

  andsteing

  authored and

  fedjax authors

  committed
  4326b85

Commits on Jul 25, 2022

• Rename jax.experimental.stax -> jax.example_libraries.stax

  Show description for 5b1dbca
  

  Jake VanderPlas

  authored and

  fedjax authors

  committed
  5b1dbca

Pagination

PreviousNext