FAX: Scalable and Differentiable Federated Primitives in JAX
Paper
•
2403.07128
•
Published
•
11
Keith Rush
jkr26
AI & ML interests
Distributed ML, autodiff, scale and speed, and private ML
Organizations
