Making the Code Runnable on CPU

#10

Please check GPU functionality. This was a quick fix tested on mac :)

Also quick formatting with ruff.

DRXD1000 changed pull request status to closed

Sign up or log in to comment