Senior Machine Learning Framework Software Development Engineer, Remote Mexico

Development of modern tools for working with AI / ML on the GPU. Implementation of algorithms for working with graphic processors. Working with GPU neural network calculations. Work with the AI/ML development team and the open-source community to analyze, develop, test, and deploy performance improvements for neural networks on GPUs with AMD ROCm, working on enabling ROCm HIPSOLVER/cuSOLVER in PyTorch