An interface used to create tooling for neuron devices on AWS Resources https://awsdocs-neuron.readthedocs-hosted.com/en/latest/general/nki/index.html https://jax-ml.github.io/scaling-book/