David Emerson issues

Results 5 issues of


                                            David Emerson

Checkpointing of a T5 model results in serialization error with new Jax 0.4.5

Flax has removed optim in favor of optax in its newest versions above 0.5.3. This means that in order to run the code in this repository, one needs to downgrade...

Extracting a small code refactor from Sana's other work

# PR Type Other # Short Description Small name and type changes. Incorporating an important change that Sana has made in her MkMMD work that could be separated out into...

Fixing the tests for Mk-MMD loss such that they still pass properly on a GPU and CPU

# PR Type Fix # Short Description Clickup Ticket(s): [Link](https://app.clickup.com/t/868a6uduu) The Tests for Mk-MMD loss were not properly handling device allocation if run on a GPU. This PR fixes that...

Changes to Support Expanded Experimentation with FedDG-GA

# PR Type Feature/Experimentation. # Short Description The changes in this PR are targeted at enabling expanded experimentation with the FedDG-GA strategy for a wider set of FL approaches. Also...

Adaptive and Fed DG-GA PFL Experimentation

# PR Type Experimentation # Short Description Clickup Ticket(s): [Link](https://app.clickup.com/t/8689g93je), [Link](https://app.clickup.com/t/8688d097n) This PR implements a set of scripts for running experiments on the Heterogenous CIFAR-10 datasets using the Adaptive pFL...