google/fedjax issues and pull requests

#309 - Drop support for Python 3.8 and 3.9

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#308 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#307 - Suppress some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#306 - [LSC] Ignore incorrect type annotations related to jax.numpy APIs

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#305 - Replace uses of `jnp.array` in types with `jnp.ndarray`.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#304 - Change `jax.numpy.NINF` reference (newly deprecated) to `-jax.numpy.inf`

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago - 1 comment

#303 - [JAX] Replace jax.numpy.NINF with -jax.numpy.inf.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#302 - Cut release v0.0.17

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#301 - Error of the Stackoverflow Tokernizer example

Issue - State: open - Opened by WilliamYi96 about 1 year ago - 1 comment

#300 - Suppress some pytype errors related to improved JAX types under pytype.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#299 - [JAX] Update some type annotations to fix some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#298 - Adding an abstraction DefaultWordTokenizer and using it in StackoverflowTokenizer.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#297 - Add Adaptive Personalized Federated Learning

Pull Request - State: closed - Opened by Lando-L over 1 year ago - 2 comments

#296 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#295 - Drop Python 3.7 support

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

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

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#293 - 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

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#292 - Fix typo in for_each_client docstring

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#291 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#290 - Cut version 0.0.16 for release

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#289 - Add support for stateful clients

Issue - State: open - Opened by Lando-L over 1 year ago - 8 comments

#288 - [NumPy] Remove references to deprecated NumPy type aliases.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#287 - Modify cross entropy loss to allow dense targets.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#286 - Adafactor optimizer

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

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

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#284 - v0.0.15

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#283 - Update various Python packaging configs

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

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

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#281 - Remove flax optim test now that flax.optim is deprecated in favor of optax https://github.com/google/flax/issues/2273

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#280 - Replace `jax.xla.DeviceArray` private type with the new public type `jax.Array`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#279 - Disable pytype import error for old stax import path

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#278 - Makes `flax.optim` import future proof.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#277 - Rename jax.experimental.stax -> jax.example_libraries.stax

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#276 - v0.0.14

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#275 - Use jax.tree_util.tree_map in place of deprecated tree_multimap.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#274 - unbiased scale for DRIVE

Pull Request - State: closed - Opened by amitport almost 2 years ago - 3 comments

#273 - Updating DRIVE algorithm to have independent rotations for each client.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#272 - s/master/main where possible

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#271 - v0.0.13

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#270 - Adding [FedProx](https://arxiv.org/abs/1812.06127) implementation at fedjax.algorithms.fed_prox

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#269 - Cut release v0.0.12

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#268 - Implement standard CIFAR-100 model in fedjax.models.cifar100

Issue - State: open - Opened by jaehunro about 2 years ago - 1 comment
Labels: enhancement, contributions welcome

#267 - No-op

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#266 - 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)

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#265 - Example logistic model for CIFAR-100 to showcase artifacts required to contribute a new model implementation to fedjax.models

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#264 - Yogi optimizer

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#263 - Updating docs

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#262 - CIFAR 100 Questions

Issue - State: closed - Opened by HanGuo97 over 2 years ago - 5 comments

#261 - Update bibtex entry to reference paper

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#260 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#259 - Cutting v0.0.11

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#258 - Support for manually modifying client/server learning rate

Issue - State: open - Opened by marcociccone over 2 years ago - 1 comment

#257 - [XLA:CPU] Relax test tolerances for tests using XLA:CPU.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#256 - Arithmetic coding of unknown distributions requires both the sequence and the histogram to be encoded. Current logic was only accounting for encoding the sequence. Adding a fix.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#255 - Cut v0.0.10 release

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#254 - Split out jittable function for arithmetic_encoding_num_bits

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#253 - Fixing a bug in the terngrad quantizer and adding tests.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#252 - Move create_token_to_ids_fn call outside of preprocess_batch so it doesn't create a tf.function for every batch for every client dataset

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#251 - Fix or silence upcoming type errors in Python 3.9.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#250 - Adding support for arithmetic coding for uniform_stochastic_quantizer

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#249 - Adding support for random rotation before quantization

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#248 - Adding Terngrad quantizer

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#247 - Implementing DRIVE algorithm.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#246 - Adding structured rotation that is used in compression.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#245 - Adding ClientId to init.py

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#244 - Adding Hadamard transform to aggregators.init.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#243 - Break out token_to_ids tokenizer function for preprocessing batches. This allows the tokenizer to be used to tokenize other examples not in a FederatedDataset.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#242 - Update compression aggregator unit test comment typos

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#241 - Support for gldv2 and inaturalist datasets

Issue - State: open - Opened by marcociccone over 2 years ago - 7 comments

#240 - Simplifying getting started example in README

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#239 - Support for haiku models with non-trainable state

Issue - State: open - Opened by marcociccone over 2 years ago - 2 comments

#238 - Fix issues in examples/:

Pull Request - State: closed - Opened by kho over 2 years ago

#237 - Problem of Quick Start in Readme.md

Issue - State: closed - Opened by Ichiruchan over 2 years ago - 2 comments

#236 - Add a tree_l2_squared function for squared L2 norm of a tree.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#235 - Bumping up version 0.0.9 for release

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#234 - Use different client datasets for cluster assignment and cluster model evaluation in HypClusterEvaluator

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#233 - A simpler yet faster Walsh-Hadamard transform

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#232 - Efficient Walsh-Hadamard transform in JAX

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#231 - Update autodocs

Pull Request - State: closed - Opened by jaehunro over 2 years ago
Labels: cla: yes

#230 - Option to clip client updates in MimeLite and clip by global norm implementation

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#229 - How to create a validation dataset?

Issue - State: open - Opened by gaseln over 2 years ago - 4 comments

#228 - Remove references of Python versions in documentations

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#227 - Bump readthedocs Python version

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#226 - Remove obsolete pytest.ini

Pull Request - State: closed - Opened by kho over 2 years ago
Labels: cla: yes

#225 - Ensure run_tests.sh runs tests without using GPUs.

Pull Request - State: closed - Opened by kho over 2 years ago
Labels: cla: yes

#224 - Ensure run_tests.sh runs tests without using GPUs.

Pull Request - State: closed - Opened by kho over 2 years ago
Labels: cla: yes

#223 - Import either jax.experimental or jax.example_libraries so that we don't need to require the latest JAX release.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#222 - Optionally depend on TensorFlow but still import at the top

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#221 - Prepare for v0.0.8 release

Pull Request - State: closed - Opened by kho over 2 years ago
Labels: cla: yes

#220 - Delay importing tensorflow instead of directly depending on it in setup

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#219 - Add back TensorFlow to the list of dependencies

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#218 - FedJax depends on TensorFlow Federated?

Issue - State: closed - Opened by davidrpugh over 2 years ago - 7 comments

#217 - Cleaning up fedjax.datasets docs and pruning unused dependencies from…

Pull Request - State: closed - Opened by jaehunro over 2 years ago
Labels: cla: yes

#216 - CIFAR-100 dataset

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#215 - Full EMNIST example does not exhibit parallelization

Issue - State: closed - Opened by gaseln over 2 years ago - 2 comments

#214 - TFFSQLiteClientsIterator for reading TFF-sourced SQLite databases for use with SQLiteFederatedDataBuilder when converting for use in FedJAX

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#213 - LZMA decompressor for TFF-sourced compressed SQLite files

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#212 - SQLiteFederatedDataBuilder.add_many for bulk insertions using [executemany](https://docs.python.org/3/library/sqlite3.html#sqlite3.Cursor.executemany) for better efficiency

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#211 - [JAX] move example libraries from `jax.experimental` into `jax.example_libraries`

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#210 - Init datasets docs page

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

GitHub / google/fedjax issues and pull requests