import numpy as np def softmax(logits): raise NotImplementedError def cross_entropy(logits, targets): raise NotImplementedError def cross_entropy_grad(logits, targets): raise NotImplementedError