alshedivat / fedpa Goto Github PK

ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-11-abcd58547ee0>](https://localhost:8080/#) in <module>()
----> 1 from federated.objectives.quadratic import *
      2 from federated.learning.algorithms import *
      3 from federated.inference.moments import *
      4 from federated.inference.optimization import *
      5 from federated.inference.sampling import *

1 frames
[/content/federated/objectives/quadratic.py](https://localhost:8080/#) in <module>()
     21 import jax.numpy as jnp
     22 import numpy as np
---> 23 from jax.api import jit
     24 from sklearn import datasets
     25 

ModuleNotFoundError: No module named 'jax.api'