bharathgs/NALU

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

Neural Arithmetic Logic Units (NALU)

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