forked from Kotlin/kotlin-jupyter-libraries
-
Notifications
You must be signed in to change notification settings - Fork 0
/
deeplearning4j-cuda.json
44 lines (44 loc) · 1.54 KB
/
deeplearning4j-cuda.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
{
"description": "Deep learning library for the JVM (CUDA support)",
"properties": {
"v": "1.0.0-beta6",
"cuda": "10.2",
"slf4j": "1.7.25",
"freemarker": "2.3.29"
},
"link": "https://github.com/eclipse/deeplearning4j",
"dependencies": [
"org.freemarker:freemarker:$freemarker",
"org.nd4j:nd4j-api:$v",
"org.nd4j:nd4j-cuda-$cuda:$v",
"org.nd4j:nd4j-cuda-$cuda-platform:$v",
"org.deeplearning4j:deeplearning4j-core:$v",
"org.deeplearning4j:deeplearning4j-common:$v",
"org.deeplearning4j:deeplearning4j-datasets:$v",
"org.deeplearning4j:deeplearning4j-nn:$v",
"org.deeplearning4j:deeplearning4j-nlp:$v",
"org.deeplearning4j:deeplearning4j-ui:$v",
"org.deeplearning4j:deeplearning4j-cuda-$cuda:$v",
"org.slf4j:slf4j-simple:$slf4j",
"org.slf4j:slf4j-api:$slf4j"
],
"imports": [
"org.nd4j.config.*",
"org.nd4j.linalg.activations.*",
"org.nd4j.linalg.api.ndarray.INDArray",
"org.nd4j.linalg.dataset.DataSet",
"org.nd4j.linalg.dataset.api.iterator.DataSetIterator",
"org.nd4j.linalg.factory.Nd4j",
"org.nd4j.linalg.learning.config.*",
"org.nd4j.linalg.lossfunctions.LossFunctions.*",
"org.deeplearning4j.eval.Evaluation",
"org.deeplearning4j.nn.conf.*",
"org.deeplearning4j.nn.conf.layers.*",
"org.deeplearning4j.nn.multilayer.MultiLayerNetwork",
"org.deeplearning4j.nn.weights.WeightInit",
"org.deeplearning4j.optimize.listeners.ScoreIterationListener",
"org.deeplearning4j.datasets.iterator.impl.ListDataSetIterator"
],
"init": [
]
}