Colab now has Lux and Reactant pre-installed. Here’s a starter script for TPUs (change the runtime, and it will automatically work on CPU or GPU)
Great! “using Optimisers” is missing at the top but otherwise superb
1 Like
A positive sign for the increased adoption of Julia in ML.
This doesn’t seem to work in gpu anymore?
I’m running on T4, Reactant.set_default_backend("gpu") returns: No GPU client found
See Issues with Reactant unable to find GPU on Colab, but CUDA.jl can - #2 by giordano and messages below.