jeremiedb

using Revise
using Flux
using Zygote
using CUDA
using Statistics: mean

################################################
# Define structs
################################################
mutable struct MyRecur{T}