issues
search
google
/
fedjax
FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.
Apache License 2.0
252
stars
41
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
[pmap no rank reduce cleanup]: Update fedjax to use indexing
#311
copybara-service[bot]
closed
2 months ago
0
[numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.
#310
copybara-service[bot]
closed
3 months ago
0
Drop support for Python 3.8 and 3.9
#309
copybara-service[bot]
closed
5 months ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#308
copybara-service[bot]
closed
5 months ago
0
Suppress some pytype errors.
#307
copybara-service[bot]
closed
1 year ago
0
[LSC] Ignore incorrect type annotations related to jax.numpy APIs
#306
copybara-service[bot]
closed
1 year ago
0
Replace uses of `jnp.array` in types with `jnp.ndarray`.
#305
copybara-service[bot]
closed
1 year ago
0
Change `jax.numpy.NINF` reference (newly deprecated) to `-jax.numpy.inf`
#304
copybara-service[bot]
closed
1 year ago
1
[JAX] Replace jax.numpy.NINF with -jax.numpy.inf.
#303
copybara-service[bot]
closed
1 year ago
0
Cut release v0.0.17
#302
copybara-service[bot]
closed
1 year ago
0
Error of the Stackoverflow Tokernizer example
#301
WilliamYi96
opened
1 year ago
1
Suppress some pytype errors related to improved JAX types under pytype.
#300
copybara-service[bot]
closed
1 year ago
0
[JAX] Update some type annotations to fix some pytype errors.
#299
copybara-service[bot]
closed
1 year ago
0
Adding an abstraction DefaultWordTokenizer and using it in StackoverflowTokenizer.
#298
copybara-service[bot]
closed
1 year ago
0
Add Adaptive Personalized Federated Learning
#297
Lando-L
closed
1 year ago
2
Silence some pytype errors.
#296
copybara-service[bot]
closed
1 year ago
0
Drop Python 3.7 support
#295
copybara-service[bot]
closed
1 year ago
0
Reduce device memory usage in for_each_client & tree_{sum,mean} by donating intermediate values
#294
copybara-service[bot]
closed
1 year ago
0
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
#293
copybara-service[bot]
closed
1 year ago
0
Fix typo in for_each_client docstring
#292
copybara-service[bot]
closed
1 year ago
0
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
#291
copybara-service[bot]
closed
1 year ago
0
Cut version 0.0.16 for release
#290
copybara-service[bot]
closed
1 year ago
0
Add support for stateful clients
#289
Lando-L
opened
1 year ago
8
[NumPy] Remove references to deprecated NumPy type aliases.
#288
copybara-service[bot]
closed
1 year ago
0
Modify cross entropy loss to allow dense targets.
#287
copybara-service[bot]
closed
1 year ago
0
Adafactor optimizer
#286
copybara-service[bot]
closed
1 year ago
0
Require sphinx>=4.3.0 so doc-building works under Python 3.10
#285
copybara-service[bot]
closed
2 years ago
0
v0.0.15
#284
copybara-service[bot]
closed
2 years ago
0
Update various Python packaging configs
#283
copybara-service[bot]
closed
2 years ago
0
Match both jax.Array and DeviceArray. jax.Array is the future jax type that will replace DA, SDA and GDA.
#282
copybara-service[bot]
closed
2 years ago
0
Remove flax optim test now that flax.optim is deprecated in favor of optax https://github.com/google/flax/issues/2273
#281
copybara-service[bot]
closed
2 years ago
0
Replace `jax.xla.DeviceArray` private type with the new public type `jax.Array`.
#280
copybara-service[bot]
closed
2 years ago
0
Disable pytype import error for old stax import path
#279
copybara-service[bot]
closed
2 years ago
0
Makes `flax.optim` import future proof.
#278
copybara-service[bot]
closed
2 years ago
0
Rename jax.experimental.stax -> jax.example_libraries.stax
#277
copybara-service[bot]
closed
2 years ago
0
v0.0.14
#276
copybara-service[bot]
closed
2 years ago
0
Use jax.tree_util.tree_map in place of deprecated tree_multimap.
#275
copybara-service[bot]
closed
2 years ago
0
unbiased scale for DRIVE
#274
amitport
closed
2 years ago
3
Updating DRIVE algorithm to have independent rotations for each client.
#273
copybara-service[bot]
closed
2 years ago
0
s/master/main where possible
#272
copybara-service[bot]
closed
2 years ago
0
v0.0.13
#271
copybara-service[bot]
closed
2 years ago
0
Adding [FedProx](https://arxiv.org/abs/1812.06127) implementation at fedjax.algorithms.fed_prox
#270
copybara-service[bot]
closed
2 years ago
0
Cut release v0.0.12
#269
copybara-service[bot]
closed
2 years ago
0
Implement standard CIFAR-100 model in fedjax.models.cifar100
#268
jaehunro
opened
2 years ago
1
No-op
#267
copybara-service[bot]
closed
2 years ago
0
Add optional CIFAR-100 preprocessing logic that matches TFF's [tff.simulation.baselines.cifar100.create_image_classification_task](https://www.tensorflow.org/federated/api_docs/python/tff/simulation/baselines/cifar100/create_image_classification_task)
#266
copybara-service[bot]
closed
2 years ago
0
Example logistic model for CIFAR-100 to showcase artifacts required to contribute a new model implementation to fedjax.models
#265
copybara-service[bot]
closed
2 years ago
0
Yogi optimizer
#264
copybara-service[bot]
closed
2 years ago
0
Updating docs
#263
copybara-service[bot]
closed
2 years ago
0
CIFAR 100 Questions
#262
HanGuo97
closed
2 years ago
5
Next