flax/examples/linen_design_test at main · Pranav-Saraswat/flax · GitHub

Name		Name	Last commit message	Last commit date
parent directory ..
attention_simple.py		attention_simple.py
autoencoder.py		autoencoder.py
dense.py		dense.py
linear_regression.py		linear_regression.py
mlp_explicit.py		mlp_explicit.py
mlp_inline.py		mlp_inline.py
mlp_lazy.py		mlp_lazy.py
tied_autoencoder.py		tied_autoencoder.py
weight_std.py		weight_std.py