Is there any plans to use ND4J

iangellove / Omega-AI

Omega-AI：基于java打造的深度学习框架，帮助你快速搭建神经网络，实现模型推理与训练，引擎支持自动求导，多线程与GPU运算，GPU支持CUDA，CUDNN。

Apache License 2.0

359 stars 51 forks source link

Open vaiju1981 opened 3 months ago

vaiju1981 commented 3 months ago

Instead of Cuda Kernels, would it be possibe to use ND4J library instead or would that support be added.

iangellove commented 3 months ago

i will thinks for your issue,but nd4j is only support Basic Operations.some model must use the special kernel for the cuda.

vaiju1981 commented 3 months ago

If there are kernels, that I can help write using ND4J, let me know, I can try and attempt those.