Training models on tiny edge devices