[R] AIoT and Federated Learning Platform

Are there any AIoT or Federated learning platforms that are recommended to help write and test/train models for edge devices? Looking for a platform that allows me to write a jupyter notebook, and simulate an environment of N edge devices - each device with unique data. Any other pointers, I'm all ears!

aiot federated learning learning machinelearning platform