[neomantra]

Perhaps check out GoMLX ("an Accelerated ML and Math Framework", there's a lot of scaffolding and it JITs to various backends. Related to that project, I sometimes use GoNB in VSCode, which is Golang notebooks [2].

[1] https://github.com/gomlx/gomlx

[2] https://github.com/janpfeifer/gonb

[eliben]

Indeed, I have a plan to publish a follow-up using GoMLX - I already have the code working and just need to clean it all up and write a post