Unverified Commit b4ecc6be authored by Matthew LeMay's avatar Matthew LeMay Committed by GitHub
Browse files

fixed typo in flax-projects readme (#12466)

parent e52288a1
......@@ -502,7 +502,7 @@ First, write a Flax module that will declare the layers and computation.
import flax.linen as nn
import jax.numpy as jnp
class MLPMoudle(nn.Module):
class MLPModule(nn.Module):
config: MLPConfig
dtype: jnp.dtype = jnp.float32
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment