All Katas - ML Katas

by: newest upvotes saves

Implementing a Graph Neural Network (GNN) Layer

this month hard (>1 hr) | network gnn graph node embedding

Implement a simple **Graph Neural Network (GNN) layer** for node classification on a small graph. The layer should aggregate features from a node's neighbors and combine them with the node's own...
Building a Neural Ordinary Differential Equation (NODE) Layer

this month 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...