equinox:JAX一站式库,提供PyTorch式神经网络语法与PyTree操作

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

分支7Tags59

项目介绍

在 JAX 中,体验优雅且易于使用的神经网络与科学计算。详细了解请访问:https://docs.kidger.site/equinox/。【此简介由AI生成】

定制我的领域
232.94 K210访问 GitHub