deepjavalibrary / deepjavalibrary/djl
Add a ND4J backend
- 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
Assessment
This issue has not been assessed yet.