jordenhill / Birdbrain

A neural network framework written in Swift.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Birdbrain

A neural network library written in Swift. The feedforward neural network is loosely based on the neural net as implemented in Neural Networks and Deep Learning. The Recurrent Neural Network is based on the implementation in Recurrent Neural Networks Tutorial, Part 2 – Implementing a RNN with Python, Numpy and Theano. The Long Short Term Memory Network is based on the LSTM network as described in A Critical Review of Recurrent Neural Networks for Sequence Learning

This isn't really for any immediate use in an iOS or OSX application at the moment, simply to practice developing and experimenting with neural nets in Swift, and using the Accelerate framework and the Metal API for computing and GPGPU functions. If you find any bugs or simply want to contribute, feel free to join in!

About

A neural network framework written in Swift.

License:MIT License


Languages

Language:Swift 91.8%Language:Metal 8.1%Language:C++ 0.1%