deepjavalibrary / deepjavalibrary/djl

Add a ND4J backend

Open
#2 7 comments 6 reactions 0 assignees View on GitHub
Dominant language
Java
Stars
4.9k
Forks
759
Avg merge
16h 15m
Merged PRs (30d)
15

Description

## Description
ND4J, the accelerated linear algebra backend that powers Eclipse DeepLearning4J, should have all the necessary features to become a proper backend for DJL.

### Will this change the current api? How?
No API changes should be necessary for the basics. If ND4J provides any opportunity an API enhancement we can discuss those as a separate issue.

### Who will benefit from this feature?
DL4J has support for more native architectures (including ARM64 (Raspberry Pi/iOS/Android) and OpenPower) so it will allow people with those devices to take advantage of DJL. DL4J supports importing Keras, Tensorflow and ONNX models so if we design this correctly we can expand the types of pre-trained models the users can use.

## References
- I created a very bare bones first pass backend that's around as complete as the TensorFlow backend here: https://github.com/wmeddie/djl/tree/add-nd4j-backend/nd4j/engine As it becomes more complete I will provide a PR.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.