Give biased_logits the same dtype as logits
#191
| Job | Run time |
|---|---|
| 24s | |
| 24s |
biased_logits the same dtype as logits
#191
| Job | Run time |
|---|---|
| 24s | |
| 24s |