Finally the CUDA addition to the same Flux code. 😅
The code itself is pretty short (shorter that original because some initialization steps stay the same and get reused) but I've commented it rather elaborately aiming at beginners new to these topics.
Does it look like too much Julia advertisement now? 😅 That never was my intention...
Finally the CUDA addition to the same Flux code. 😅
The code itself is pretty short (shorter that original because some initialization steps stay the same and get reused) but I've commented it rather elaborately aiming at beginners new to these topics.
Does it look like too much Julia advertisement now? 😅 That never was my intention...