search

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX
Apache License 2.0
141 stars 9 forks source link

NN Layers: Convolutions #23

Closed dlwh closed 9 months ago

dlwh commented 11 months ago

See Equinox's stuff for basic outline: https://docs.kidger.site/equinox/api/nn/conv/