Software Engineer, Model Inference

Job summary

Responsibilities

We are looking for an engineer who wants to take the world's largest and most capable AI models and optimize them for use in a high-volume, low-latency, and high-availability production and research environment.

• Work alongside machine learning researchers, engineers, and product managers to bring our latest technologies into production.
• Work alongside researchers to enable advanced research through awesome engineering.
• Introduce new techniques, tools, and architecture that improve the performance, latency, throughput, and efficiency of our model inference stack.
• Build tools to give us visibility into our bottlenecks and sources of instability and then design and implement solutions to address the highest priority issues.
• Optimize our code and fleet of Azure VMs to utilize every FLOP and every GB of GPU RAM of our hardware.

Tech stack

• Azure
• PyTorch
• CUDA