Dilshod Tadjibaev

All changes are in place. Waiting for @laggui. If you have a chance, @wandbrandon, please review as well.

@laggui Done updating. Hopefully no issues. Thanks for reviewing the long PR. Takes up lots of context switching.

> Hi, could I take up this issue? Yes! Please go ahead. We would appreciate your contribution. Let me know if you need more info.

> It seems that this issue is still open and there hasn't been any updates, I would love to take up the issue if possible. > > Just to clarify,...

@nwhitehead, @med1844 thanks for your inputs. Yes, I agree with you. I came to the same conclusion about the need to verify if a ported model matched the original model....

Just keep in mind there is web mnist example that uses the output of mnist training. I optimized the model for web.

There hasn't been activity in this PR. Closing it for now. Please feel to reopen or create a brand new PR.

@nathanielsimard @louisfd just be aware of this jit related bug on WASM/WebGPU

I would be great if we had WebGPU tests on CI #810

I believe this is already implemented as of burn 0.17