基于 JAX 的计算机视觉研究库,提供注意力机制、Transformer 和 Vision Transformer 等模型实现,方便快速搭建和复现视觉研究实验。
基于 PyTorch 的知识蒸馏框架,无需写代码即可复现和对比 28 种顶会蒸馏方法,附带训练模型、日志和配置,方便做可复现的深度学习研究。