EfficientNet JAX-亚麻亚麻和欧柏克斯 介绍 这是一项巨大的蒸蒸工作。 我正在使用的Jax，jaxlib和NN库每周都在转移。 该代码库当前支持： 亚麻亚麻（ 用于模型，带有预训练重量的验证以及从头开始训练 Objax（ ）-使用预先训练的权重进行模型和模型验证 这本质上是我的PyTorch EfficienNet生成器代码（ 的改编，也可以在找到）改编成JAX。 我开始这个 通过以熟悉的代码/模型作为起点来学习JAX， 找出我喜欢的JAX建模接口库（“框架”）， 在P