redco-v0.4.20
·
40 commits
to master
since this release
- Updated data example type support -- can be a
list
of whatever types now, e.g.,examples=[str, str, str, ...]
orexamples=[dict, dict, dict, ...]
- Updated mixed-precision training -- by setting
compute_dtype
, e.g.,Trainer(compute_dtype=jnp.bfloat16)
.