c0dearm / mushin

Computational graphs with reverse automatic differentation in the GPU
Other
387 stars 9 forks source link

Note for followers! #23

Open c0dearm opened 2 years ago

c0dearm commented 2 years ago

Mushin is undergoing a major refactor, in order to provide a better API (experience) to users creating their own Deep Learning models. During the implementation of the MNIST digit recognition example I found some difficulties to ergonomically express the convolutional model. You might not see commits often until I iterate over a few different implementation possibilities. Until then, just letting you know I am working on it!

Thank you all for your patience :heart: