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 somet

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 https://github.com/gomlx/gomlx [2] https://github.com/janpfeifer/gonb https://github.com/janpfeifer/gonb