viva_tensor/training_demo

Types

pub type TrainingState {
  TrainingState(
    tape: autograd.Tape,
    layer1: nn.Linear,
    layer2: nn.Linear,
  )
}

TrainingState(
  tape: autograd.Tape,
  layer1: nn.Linear,
  layer2: nn.Linear,
)

</>

pub fn main() -> Nil

✨