Main (Seonghyeon)
- control timesteps
- initialize everything
Node (2 people) (Ethan, Yuting)
- Multiply matrices
- Calculate gradients
ParameterServer (Andrew)
- take average
- apply gradient to average of medians
Batch (Sohyun)
- Calculate geometric median
We could consider using this library https://deeplearning4j.org