3D Turbulence - Pseudo-spectral DNS Simulation in Python - Taylor-Green Vortex

Pseudo-spectral DNS simulation made in python. The code used was based on Mortensen, M. and Langtangen, H. P. (2016) methodology for 3D high performance turbulence simulations.
Some parameters of the simulation were:
Reynolds Number=1600;
Total Physical Time: 20;
Time Step: 0.05;
Physical Box: [0, 2*pi]^3;
Grid Size: 128^3;
Initial Conditions: Velocity field initialized with the following initial conditions: u=sin(x)*cos(y)*cos(z), v=-cos(x)*sin(y)*cos(z) and w=0;
Boundary Conditions=Periodic.
The simulation was conducted on a single CPU using Google Colab.