> I would guess the constant memory allocation and frees in the training loop are the bottleneck
No, the bottleneck would be not utilizing the idling GPU.
No, the bottleneck would be not utilizing the idling GPU.
> simple, purpose written NNs for many simple applications [... as opposed to] python and cuda libraries