fedjax (pypi): Federated learning simulation with JAX.
Keywords: federated|python|machine|learning|federated-learning|jax