FastAI and Sagemaker EI

So because I need to add an extra level of difficulty to everything I do, it smells like it should be possible to get FastAI to use Sagemaker Elastic Inference now: https://www.datanami.com/2020/03/23/aws-announces-support-for-pytorch-with-amazon-elastic-inference/ and https://github.com/aws/sagemaker-pytorch-serving-container. Though as I’m learning Jupyter, I’m not sure which kernel to use and whether I should be installing all this missing packages after I switch kernels. Has anyone got this working yet? I’m Linux & bash savvy, so happy to do a little grunt work.