def METHOD_NAME(input: dc.float32[N, C_in], w1: dc.float32[C_in, S0], b1: dc.float32[S0], w2: dc.float32[S0, S1],
               b2: dc.float32[S1], w3: dc.float32[S1, S2], b3: dc.float32[S2]):
    x1 = relu(input @ w1 + b1)
    x2 = relu(x1 @ w2 + b2)
    x3 = softmax(x2 @ w3 + b3)  # Softmax call can be omitted if necessary
    return x3