Profound ask to help in accelerating counting the epochs.

[AdrianRobak]

Dear Rémi Louf,

I'm writing to you because I have a problem with my project:
https://github.com/AdrianRobak/Biaxial3.git

I want to ask you about GPU and TPU as an option in Colab Pro, AWS, Google Could, Genesis Cloud, and more, which I recently used to accelerate the counting of epochs.
Could you help me? Because during counting LSTM neural network in my project "Biaxial3," neither Colab's GPU nor TPU didn't accelerate it. My neural network is built in Python 3.11 and uses the Aesara library. I thought every counting would be faster, but in my case, it was even slower than in the internal GPU. Normal session in my computer (2019 iMac) continues for approximately four days, and in Colab Pro with GPU/TPU, eight days.
I tried to accelerate counting my LSTM neural network using GPU A100, V100, T4, and TPU in Colab and other units in AWS (like g3s.xlarge $0.938, g3.4xlarge $1.425, g4dn.xlarge $0.658 and inf1.6xlarge) but these didn't work :(. I tried other computing companies. Unfortunately - these failed. I expected my LSTM neural network to count and finish in 4 hours rather than 4 or 8 days. I have an account in Colab and lost a lot of units to watch how GPU and TPU didn't work. I tried running another repository in Colab, but it was built using Tensorflow and Keras libraries, and it worked adequately, counting fast.

"Biaxial3" is my repository which I upgrade from another guy in GitHub. The results could be exciting in musicology research, so I need your help. Could you help me accelerate it? I would appreciate any help and resolve my problem.
I'm not an IT worker, but I'm a composer :), so I am trying to figure out what to do now, especially since I've never really learned about coding (I have participated in only one course about machine learning). I was determined to know and understand something in Python. I am scared that the code inside the file "model.py" and the other four files must be changed, but I lack specialized knowledge and skills.

Could you help me accelerate it?
I would appreciate any help and resolve my problem.

Adrian