Scalable HW-Aware Training for Analog In-Memory Computing - View it on GitHub