This is an automated archive made by the Lemmit Bot.

The original was posted on /r/opensource by /u/LlaroLlethri on 2024-02-12 11:28:46.

It started as a personal effort to learn more about neural networks, but I got a bit carried away and kept refining it and adding more features :)

I recently got it running on the GPU with Vulkan compute shaders 🎉

It’s a CLI app where you just describe the shape of your network in a json config file and point it at your training data, then run it again in evaluation mode and point it at some data it hasn’t seen before and it will try to classify each sample.

If you’re interested in neural nets, you’ll hopefully find the code quite readable, and I’ve written up the math in documentation/math.pdf.

It’s very much a work in progress at the moment. I have a whole backlog of features I’d like to add, so watch this space :)