- Wed 20 December 2023
- Neuron
What is Neuron SDK
This post introduces AWS Neuron SDK - an SDK that streamlines deep learning and generative AI workloads on AWS Inferentia and Trainium by integrating with frameworks like PyTorch and JAX.
Musings of a machine learning researcher, engineer and leader