Skip to content

(currently incredibly naive) cuda deep learning library

Notifications You must be signed in to change notification settings

knightron0/lattice

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

46 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

(currently incredibly naive) cuda deep learning library

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published