kerax:JAX框架的类似Keras的API 源码

凯拉克斯 用于库的类似Keras的API。 产品特点 进行高性能的机器学习研究。 内置支持流行的优化算法和激活功能。 在CPU,GPU甚至TPU上无缝运行! 无需任何手动配置。 快速开始 码 from kerax . datasets import binary_tiny_mnist from kerax . layers import Dense , Relu , Sigmoid from kerax . losses import BCELoss from kerax . metrics import binary_accuracy from kerax . models import