Deep learning benchmark for buddy-benchmark [WIP]