Full threadJugglerofworlds·I recently made a similar implementation for SHA256. Unfortunately it suffers from gradient vanishing. Source code: https://github.com/calebh/differentiablesha256View on HN