/NALU

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units paper by trask et.al

Primary LanguagePythonMIT LicenseMIT

Neural Arithmetic Logic Units (NALU)

Downloads GitHub Hackage-Deps PyPI - Python Version

Basic pytorch implementation of NAC/NALU from Neural Arithmetic Logic Units by trask et.al

Installation

pip install NALU

Usage

from nalu.core import NaluCell, NacCell
from nalu.layers import NaluLayer