Add a simple MLP module

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

140 stars 9 forks source link

Closed dlwh closed 8 months ago

dlwh commented 8 months ago

MLPs aren't the most Haliax-friendly module type, but it's not too bad.