/lattice

(currently incredibly naive) cuda deep learning library

Primary LanguageCuda

lattice

some kind of an ml library in cuda

rough timeline:

  • feedforward neural network
  • manual gradient calculation for backward pass
  • automatic gradient calculation
  • add convolution expression
  • convolutional neural network

...some intermediate steps, then eventually:

  • transformer

somewhere along the line:

task for initial steps: MNIST dataset