MLP(  # 131 params, 524 B
  activation=relu,
  # Modules:
  layer_1=Linear(  # 80 params, 320 B
    # Parameters:
    w=Param(f32(4, 16)),  ▅▅▅▄▅▅▅▄▂▂█  μ=-0.0092 σ=0.31
    b=Param(f32(16,)),    ▁▁▁▁▁█▁▁▁▁▁  μ=0 σ=0
  ),
  layer_2=Linear(  # 51 params, 204 B
    # Parameters:
    w=Param(f32(16, 3)),  ▅▅▄▅▅▅▅█▃▅▅  μ=0.028 σ=0.31
    b=Param(f32(3,)),     ▁▁▁▁▁█▁▁▁▁▁  μ=0 σ=0
  ),
)