Skip to content

redco-v0.4.20

Compare
Choose a tag to compare
@tanyuqian tanyuqian released this 22 Jul 01:42
· 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, ...] or examples=[dict, dict, dict, ...]
  • Updated mixed-precision training -- by setting compute_dtype, e.g., Trainer(compute_dtype=jnp.bfloat16).