elegy:Elegy是Jax生态系统的与框架无关的Trainer界面 源码

挽歌 Elegy是Jax生态系统的与框架无关的Trainer界面。 主要特点 易于使用:Elegy提供了类似Keras的高级API,可以很轻松地完成常见任务。 灵活:Elegy提供了一个功能类似于Pytorch Lightning的低级API,在需要时提供了最大的灵活性。 不可知论:Elegy支持各种框架,包括高级API上的Flax,Haiku和Optax,而对于低级API则100%与框架无关。 兼容:Elegy可以使用多种常见数据源,包括TensorFlow数据集,Pytorch数据加载器,Python生成器和Numpy pytree。 有关更多信息,请参阅“ 。 安装 使用pip安装Elegy: pip install elegy 对于Windows用户,我们推荐用于linux 2 的Windows子系统,因为还不支持它。 快速入门:高级API Elegy的高级API提供了一个非