MLP(  # 131 params, 524 B, 80 frozen
  activation=relu, final_activation=None,
  # Modules:
  (0): Linear(  # 80 params, 320 B, 80 frozen
    # Parameters:
    w=Param(f32(4, 16), frozen),  ▅▅▅▄▅▅▅▄▂▂█  μ=-0.021 σ=0.69
    b=Param(f32(16,), frozen),    ▁▁▁▁▁█▁▁▁▁▁  μ=0 σ=0
  ),
  (1): Linear(  # 51 params, 204 B
    # Parameters:
    w=Param(f32(16, 3)),  ▅▅▄▅▅▅▅█▃▅▅  μ=0.031 σ=0.34
    b=Param(f32(3,)),     ▁▁▁▁▁█▁▁▁▁▁  μ=0 σ=0
  ),
)