Machine Learning for Artificial Intelligence
机器学习,人工智能,张量计算库。致力于开发一个基于java语言的深度学习工具箱,项目是基于Nd4j做运算,可以使用GPU加速大矩阵。支持自动微分、动态计算图等。 开发这个的目的在于结合java与pytorch的优势,目前来说还没有一个很易于编程的基于java的自动微分张量计算图工具,本工具就是为了此编写。
依赖张量运算库:Nd4j 可选的库:dl4j
1、提供了实例,以实例来做为教程,请原谅没有时间做视频或者专业资源,提供了实例来说明。
动量实例:com.ml4ai.demo.MomentTest
自动微分:com.ml4ai.demo.AutoGrad
MLP分类:com.ml4ai.demo.MLPClassification
拟合kx plus b:com.ml4ai.demo.Demo
请联系: sleechengn@163.com ,加入力量,这是一个持续开发的项目,希望大家为java深度学习做贡献。
网站 :www.ml4ai.com