/SwanKit

Deep learning framework in Swift (& Metal)

Primary LanguageSwiftMIT LicenseMIT

SwanKit

SwanKit is a framework for programming neural networks inspired by Torch.

It attempts to make deep learning more accessible for regular Mac users.

Initial goal is to implement it using Metal Framework for Mac and iOS. A long-term goal is to use CUDA for other platforms.

If you are interested to contribute, please check here.

Build Status

Using SwanKit

Tensors and operations

To create a tensor use SWKTensor class:

import SwanKit

let a = SWKTensor(2, 2)
let b = SWKTensor(content: [[1, 2], [3, 4]])

TODO: complete

Computational graph and differentiation

TODO: complete

Learning routines

TODO: complete