Star AlbumentationsX on GitHub — it powers this leaderboard
tensorflow/graphics
TensorFlow Graphics: Differentiable Graphics Layers for TensorFlow