bobbin (pypi): Tools for making training loops with flax.linen models.
Keywords: python|machine|learning|flax