search

stanford-crfm / haliax

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

Make Conv work when input type is not the same as kernel type #78

Open dlwh opened 3 months ago

dlwh commented 3 months ago

We should either do widening or, once #65 is merged, use that when present