JAX
JAX深度学习官网,Google开源的一个用于机器学习和科学计算的Python库GitHub is where over 100 million developers shape the future of software, together. Contribute to the open source community, manage your Git repositories, review code like a pro, track bugs and features, power your CI/CD and DevOps workflows, and secure code before you commit it.