grain

Modules

autograd
module grain.autograd
chain
module grain.chain

Chain means autograd operators in grain that is equivalent to - pytorch: torch.nn.Module - chainer: chainer.Chain or chainer.Link

cublas
module grain.cublas
cudnn
module grain.cudnn
functions
module grain.functions
testing
module grain.testing
utility
module grain.utility