OpenCL accelerated nd-arrays / tensors for Java, Kotlin, Groovy, Scala, Jython, JRuby...
Current Build | Code Coverage | Version | Code Quality | Licence | Size |
---|---|---|---|---|---|
|
|
Impress me! | Show me more | Documentation |
---|---|---|
|
Neureka trains your neural network using a computation graph recorder.
This is contrary to the approaches found in other frameworks such as TensorFlow, Theano, Caffe, and CNTK which require the definition of a computation graph ahead of time. This means a developer has to build a neural network structure which cannot change during runtime.
Neureka, uses the recorded computation graph in order to apply a technique called reverse-mode auto-differentiation,
which allows your network structure to change during runtime arbitrarily with zero lag or overhead.
This powerful feature has been inspired by PyTorch:
- Motivation π₯
Package | Description |
---|---|
neureka | the root package containing the tensor class and the following sub-packages |
neureka.devices | a sub-package which enables cross platform acceleration (OpenCLDevice ) and tensor persistence (FileDevice ) |
neureka.math | a sub-package containing collections of functions and the ability to create custom ones |
neureka.optimization | a sub-package for weight-gradient optimization |
neureka.autograd | the guts of Neurekas autograd system |
neureka.backend | the backend containing both a consistent API and a standard implementation |
<dependency>
<groupId>com.github.gleethos</groupId>
<artifactId>neureka</artifactId>
<version>1.0.1</version>
</dependency>
Groovy DSL:
implementation 'com.github.gleethos:neureka:1.0.1'
Kotlin DSL:
implementation("com.github.gleethos:neureka:1.0.1")
1. Add the JitPack url in your root build.gradle
at the end of repositories
allprojects {
repositories {
//...
maven { url 'https://jitpack.io' }
}
}
2. Add Neureka as dependency
...either by specifiying the version tag:
dependencies {
implementation 'com.github.Gleethos:neureka:v1.0.1'
}
...or by using a custom commit hash instead:
dependencies {
implementation 'com.github.Gleethos:neureka:8485bca'//Any commit hash...
}
@GrabResolver(name = 'jitpack.io', root = 'https://jitpack.io')
@Grab('com.github.Gleethos:neureka:v1.0.1')
import neureka.*
Execute the following:
gradlew build
Tests:
gradlew check
Jar file:
gradlew jar
Min-jar file:
gradlew proguard
- JOCL 2.+ - (OpenCL binding)
- SLF4J 1.7.+ - (Logging API allowing for custom backends)
- BDD & living documentation with Spock!
- Yes! A browsable test suite. Check it out!!
Want to contribute? Don't worry:
There is plenty of developer friendly highly readable living documentation to go through which explains the inner and outer workings of this project very well!
If you want to dive right into it, start off by extending the backend for additional types of operations or data type support.
Any feedback or contribution, even as simple as a typo fix, is always greatly appreciated!
- Make a wish! :)
- MIT -> It's Free! ...