NMS and gradient complete
Test Benches not done yet