nn_forward.rs

// Create a new neural network with 2 input features, 8 hidden units in each of the two hidden layers, and 1 output feature
let mut nn = NN::new(2, 8, 8, 1, &mut rng);
// ...
// Perform a forward pass through the neural network to compute the predicted output
let y = nn.forward(&x);