苹果开源基于Jax的深度学习库AX
talkingdev • 2023-07-21
1228573 views
近日，苹果公司发布了一款新的深度学习库AX，该库基于Jax构建。值得一提的是，AX支持基于编译器的并行性，可实现大规模训练，这是许多其他流行框架所不具备的功能。在过去的几年中，深度学习领域的研究和应用取得了飞速发展，各类深度学习框架应运而生。AX库的发布意味着苹果在人工智能领域的布局愈发明确。Jax是谷歌开源的一款用于高性能机器学习研究的库，通过支持基于编译器的并行性，AX库在大规模训练方面具有明显优势。这将有助于提高AI训练效率，从而促进人工智能在各行业的广泛应用。AX库已在GitHub上开源，欢迎广大开发者和研究者参与其中，一起推动深度学习技术的发展。