search

stanford-crfm / haliax

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

better exception when calling .item inside jit #58

Open dlwh opened 5 months ago

dlwh commented 5 months ago

people keep getting bit by this