ptoulme-aws

Machine Learning Engineer II - Annapurna Laboratories Amazon Web Services

Annapurna Laboratories - Amazon Web Services

Pinned Repositories

axlearn
Language:Python14
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python30.4k2.8k
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++2.7k432
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Language:C++2.5k480
neuronx-nemo-megatron
Language:Python00
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++00
axlearn
An Extensible Deep Learning Library
Language:Python1.9k264
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
Language:Python459153
neuronx-nemo-megatron
Language:Python3211
aws-neuron-parallelcluster-samples
Language:Shell226

ptoulme-aws's Repositories

ptoulme-aws/axlearn
Language:Python14
ptoulme-aws/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++
ptoulme-aws/neuronx-nemo-megatron