issues
search
stanford-crfm
/
haliax
Named Tensors for Legible Deep Learning in JAX
Apache License 2.0
141
stars
9
forks
source link
Add more NN Layers
#22
Open
dlwh
opened
11 months ago
dlwh
commented
11 months ago
I punted on adding a bunch of different layers:
[x] #23
[x] TransposeConvolutions
[ ] #24
[ ] BatchNorm (ugh, stateful)
[x] #25
[ ] #26
[ ] #27
[x] RMSNorm
[ ] SpectralNorm?
[ ]
I punted on adding a bunch of different layers: