Custom data types and layouts for training and inference - View it on GitHub