ptoulme-aws

Machine Learning Engineer II - Annapurna Laboratories Amazon Web Services

Annapurna Laboratories - Amazon Web Services

Pinned Repositories

axlearn
An Extensible Deep Learning Library
Language:Python1.9k 63 16269
aws-neuron-parallelcluster-samples
Language:Shell22 27 86
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:Python464 74 847154
neuronx-nemo-megatron
Language:Python33 26 511
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Language:Python30.6k 337 5.7k2.8k
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++2.7k 41 371436
axlearn
Language:Python1 0 05
neuronx-nemo-megatron
Language:Python00
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Language:C++00
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Language:C++2.5k 58 2.3k483

ptoulme-aws's Repositories

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