Awesome Open Source
Awesome Open Source

Komputation

Komputation is a neural network framework for the Java Virtual Machine written in Kotlin and CUDA C.

Maven

Komputation is available through Maven Central:

<dependency>
    <groupId>com.komputation</groupId>
    <artifactId>komputation</artifactId>
    <version>0.12.5</version>
</dependency>

Layers

CPU demos

GPU/CUDA demos

Sample code

The following code instantiates a GPU-accelerated convolutional neural network for sentence classification:

    val sentenceClassifier = cudaNetwork(
        batchSize,
        lookup(embeddings, maximumDocumentLength, embeddingDimension, optimization),
        convolution(numberFilters, filterWidth, filterHeight, initialization, optimization),
        relu(),
        dropout(random, keepProbability),
        dense(numberCategories, Activation.Softmax, initialization, optimization)
    )

See the TREC demo for more details.

Initialization

Loss functions

Optimization


Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
kotlin (3,754
machine-learning (3,721
framework (1,127
nlp (1,116
artificial-intelligence (666
convolutional-neural-networks (457
neural-networks (446
gpu (376
cuda (366
jvm (215
recurrent-neural-networks (148
nvidia (121
seq2seq (107