This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commits

Commits on Jun 17, 2020

Commits on May 21, 2020

add @noDerivative to isReducedPrecison declaration (#963 )
mikowals
authored

Commits on Feb 12, 2020

Fix README: backprop returns a tuple (#666 )
mikowals
authored

Commits on Aug 19, 2019

add test for Conv2D gradients (#443 )

mikowals
authored and
rxwei
committed

Commits on Aug 13, 2019

Fix LayerNorm test and modify scale and offset broadcast to match Python (#440 )

mikowals
authored and
rxwei
committed

Commits on Aug 10, 2019

add test for BatchNorm inference (#438 )

mikowals
authored and
saeta
committed

Commits on Jul 24, 2019

use .moments() in LayerNorm and BatchNorm layers (#384 )

mikowals
authored and
rxwei
committed

Commits on Jul 22, 2019

add __owned in _vjpScalarInit (#379 )

mikowals
authored and
rxwei
committed

Commits on Jul 11, 2019

Fixed the gradient of 'Tensor.rsqrt()' (#350 )

mikowals
authored and
eaplatanios
committed

Commits on Jul 1, 2019

Add orthogonal initializer (#312 )

mikowals
authored and
rxwei
committed