MNIST CNN - CPU vs GPU

This project benchmarks CNN inference on MNIST across CPU (PyTorch/JIT) and GPU (CUDA), including NCHW vs NHWC layout analysis.

CUDA vs PyTorch throughput and memoryExecution time breakdownMemory usage comparison

Paper

GPU/CPU architecture overview and results.