I think a js framework doesn’t make sense because all people I know dealing with dnns using python as base language. I’m also working on something very similar

I think a js framework doesn’t make sense because all people I know dealing with dnns using python as base language. I’m also working on something very similar and I think my stack is a lot better because I used pyodide + webassembly + webgpu . So far, I managed to implement a very small grad based dnn framework with couple of ops. (like matmul, etc.) One of the limitations is that webassembly vm has 4gb address space. This could be a problem during loading/unloading from/to gpu.