Algorithms for Privacy-Preserving Machine Learning in JAX - View it on GitHub