Checkout the README on GitHub! It has an example split.
Honestly the WASM part was pretty straightforward, other than some shader performance issues with Burn + CubeCL , but nothing this project can really change.
The hardest part was the weight porting honestly, since I had to perfectly recreate all of the digital signal processing and model, any deviations from the original PyTorch implementation, and you ended up with subtle loss of signal in the outputs. Added that folder `bench/` to step through the two implementations piece by piece to help debug where they were diverging -- and it's still not a perfect replication of the original after all the tuning!
Honestly the WASM part was pretty straightforward, other than some shader performance issues with Burn + CubeCL , but nothing this project can really change.
The hardest part was the weight porting honestly, since I had to perfectly recreate all of the digital signal processing and model, any deviations from the original PyTorch implementation, and you ended up with subtle loss of signal in the outputs. Added that folder `bench/` to step through the two implementations piece by piece to help debug where they were diverging -- and it's still not a perfect replication of the original after all the tuning!