Machine Learning Engineer obsessed with ML Systems, Inference Optimization, and PyTorch internals. Here I am documenting everything I make or work on.. soo yeah.
Deconstructing my PyTorch ATen PR: What happens when you index a Tensor?
How I fixed a crash in PyTorch’s ATen C++ layer when tensor indexing receives an empty index list — tracing the dispatch from Python through to IndexingUtils.h and TensorAdvancedIndexing.cpp.