Contains all deep learning related code