fedpa

eg3d

fedhf

🔨 A Flexible Federated Learning Simulator for Heterogeneous and Asynchronous.

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs....

tensorflow-federated

An open-source framework for machine learning and other computations on decentralized data.

stanford_alpaca

Code and documentation to train Stanford's Alpaca models, and generate the data.

dfgo

Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021

stylegan3

Official PyTorch implementation of StyleGAN3

CLsurvey

Continual Hyperparameter Selection Framework. Compares 11 state-of-the-art Lifelong Learning meth...

fidjax

Frechet Inception Distance in JAX

alpa

Training and serving large-scale neural networks with auto parallelization.

FedNAS

FedNAS: Federated Deep Learning via Neural Architecture Search

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

edm

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

distributed_learning_simulator

For distributed machine learning