**Task Description:** [The code is about 300 lines in Python, and I can run it on a GPU for testing. The implementation is mostly complete, but there are a few bugs that need to be resolved.] I am looking... (Budget: $30 - $250 USD, Jobs: Artificial Intelligence, Machine Learning (ML), Pytorch, Tensorflow)