Pytorch?

[watts4speed]

Does a pytorch version of the algorithm exist?

[watts4speed]

I think I found one.

https://github.com/emptydiagram/ngc

[ago109]

Oh this is cool - thanks for pointing to this PyTorch implementation of the original NGC models =)

In terms of ngc-learn, the general neuroscience library (the spiking neural networks, general predictive coding, other NeuroAI models, etc.), currently it is written with JAX in mind. However, there was chat amongst the dev-team to talk about potential PyTorch support/interfacing or PyTorch-like APIs at least.
I will ping/notify @rxng8, as this dev had, at one point, whipped up a mock/prototype of what a PyTorch-like variant of ngc-learn might look like. @rxng8 do you have any comments to provide on this point?

[watts4speed]

Adding a bit more, hoping to not be to obvious: adding pytorch would over the long run, open up your library to a ton of potential users. Including me :-)

…

On Sun, Feb 22, 2026 at 2:12 PM Alex Ororbia ***@***.***> wrote: *ago109* left a comment (NACLab/ngc-learn#141) <#141 (comment)> Oh this is cool - thanks for pointing to the PyTorch implementation of the original NGC models =) In terms of ngc-learn, the general neuroscience library (the spiking neural networks, general predictive coding, other NeuroAI models, etc.), currently it is written with JAX in mind. However, there was chat amongst the dev-team to talk about potential PyTorch support/interfacing or PyTorch-like APIs at least. I will ping/notify @rxng8 <https://github.com/rxng8>, as this dev had, at one point, whipped up a mock/prototype of what a PyTorch-like variant of ngc-learn might look like. @rxng8 <https://github.com/rxng8> do you have any comments to provide on this point? — Reply to this email directly, view it on GitHub <#141 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AA4D5JK6X5V2B4HAO3YVUNL4NISVNAVCNFSM6AAAAACV3N3AIGVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZTSNBRHAYDINRVGI> . You are receiving this because you authored the thread.Message ID: ***@***.***>