Setup Nx lib and EXLA to run NX/AXON with CUDA
Steps for you setup and run Machine Learning with Axon or simple Nx script with EXLA on CUDA (GPU). Setup CUDA on local machine. For case you want to try run model on GPU (Linux/Ubuntu) you need setup CUDA environment...