Pinned Repositories
aws-neuron-sdk
Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
neuronx-nemo-megatron
aws-neuron-sdk
AWS Neuron SDK
axlearn
An Extensible Deep Learning Library
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
axlearn
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
HahTK's Repositories
HahTK/aws-neuron-sdk
AWS Neuron SDK
HahTK/axlearn
An Extensible Deep Learning Library