search

lucpaoli / SAFT_ML

2 stars 0 forks source link

multi-threaded training #7

Closed lucpaoli closed 10 months ago

lucpaoli commented 11 months ago

Two points it could be multithreaded:

lucpaoli commented 11 months ago

Problem: Make sure no problems with race conditions etc?

lucpaoli commented 11 months ago

a_res now works with Zygote, suggesting could be GPU-compatible. Will need to properly check.

lucpaoli commented 10 months ago

Resolved, see the testcases for Zygote https://github.com/FluxML/Zygote.jl/blob/master/test/threads.jl