Type-safe, high performance, distributed Neural networks in Scala (not Python, finally...).
Low level (linear algebra) operations powered by low level TensorFlow API (C, C++ bindings via JNI).
Scala used to build computation graphs and compile them into native tensor graphs.
Compiled graphs are fully calculated in native code (on CPU, GPU or TPU)
and only result is returned back via DirectBuffer
which points into native memory.
DirectBuffer
is wrapped with Tensor
object which allows
to slice and read data in a convenient way (just like Breeze
or Numpy
does).
todo
DONE
- Enhance Session
- Add basics ops
- Handle tensor graphs
- Add gradient support to ops
- Add the rest of the ops
- todo
- todo
- todo
- todo
- todo
- todo
- todo