All Katas - ML Katas

by: newest upvotes saves

Building a Neural Ordinary Differential Equation (NODE) Layer

this year hard (>1 hr) | node ode continuous differential equation

Implement a simple **Neural Ordinary Differential Equation (NODE) layer**. This involves defining a `torch.nn.Module` that represents the derivative function $f(t, z(t))$ and then using...