add hypothesis tests (mostly testing shapes)

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

141 stars 9 forks source link

Open dlwh opened 11 months ago

dlwh commented 11 months ago

We have decent-ish test coverage but we don't test the full set of weird shapes and things. would be good to add hypothesis tests.