title: Data augmentation on GPU
body: <p>As tf.data augmentations are executed only on CPUs. I need a way to run certain augmentations on the TPU for an audio project.<br />
  For example,</p>
  <blockquote>
  <p>CPU: tf.recs read -&gt; audio crop -&gt; noise addition.<br />
  TPU: spectogram -&gt; Mixup Augmentation.</p>
  </blockquote>
  <p>Most augmentations can be done as a Keras Layer on top of the model, but MixUp requires both changes in input as well as label.</p>
  <p>Is there a way to do it using tf keras APIs.</p>
  <p>And if there is any way we can transfer part of tf.data to run on TPU that will also be helpful.</p>