Hacker News new | ask | show | jobs
by brucethemoose2 1153 days ago
Also I bet the pytorch training code is written with CUDA semantics. Maybe a JAX version would work without messing with the code.