Who We Are

class AIModel:
  def __init__(self):
    self.layers = []
    self.learning_rate = 0.001

  def predict(self, x):
    for layer in self.layers:
      x = layer.forward(x)
    return x

  def train(self, x, y):
    # Forward pass
    output = self.predict(x)

    # Backward pass
    grad = self.loss_fn.backward(output, y)
    for layer in reversed(self.layers):
      grad = layer.backward(grad)