Custom Operations# Create a Custom Operator in C++ Prerequisites Operator Definition CPU Operator Implementation GPU operator implementation Building the Plugin Importing the Plugin Preloading the plugins Python Operators Defining an Operation Defining a Pipeline Running the Pipeline and Visualizing the Results Variety of Python Operators Processing GPU Data with Python Operators CuPy Operations Defining a Pipeline Running the Pipeline and Visualizing the Results Advanced: Device Synchronization in the DLTensorPythonFunction Numba Function - Running a Compiled C Callback Function Define the shape function swapping the width and height Define the processing function that fills the output sample based on the input sample Define the Pipeline Running custom JAX augmentations in DALI Setting up the example Adding an augmentation defined with JAX JAX function transformations Multiple inputs and outputs JAX augmentations in regular pipelines Running JAX augmentations on multiple GPUs Setting up distributed run Iterator with sharding JAX augmentation with sharding