GPU Memory Allocation for Machine Learning