All Katas - ML Katas

by: newest upvotes saves

Build a Simple Neural Network with Flax

yesterday medium (<30 mins) | mlp jax flax neural-network

Using Flax, JAX's neural network library, build a simple Multi-Layer Perceptron (MLP). The MLP should have an input layer, one hidden layer with a ReLU activation function, and an output layer....
Implement a Simple MLP

this month medium (<30 mins) | mlp nn forward pass

Build and run a minimal Multi-Layer Perceptron (MLP) using `torch.nn`. ### Problem Construct a 2-layer MLP with ReLU activation for input of size 10 and output of size 2. - **Input:** Tensor of...