Solution Architect (Kernel Optimization & ML Performance)

Responsibilities

• Define and own the architecture for performance-critical ML workloads, leveraging custom kernels on TPUs and GPUs
• Create a strategic roadmap for kernel optimization, framework integration, and large-scale performance improvement
• Collaborate with the client’s technical leadership, ML researchers, and engineers to capture requirements and design scalable solutions
• Evaluate and select the right technologies, toolchains, and design patterns to optimize compute-intensive operations
• Guide the development of benchmarking infrastructure, autotuning frameworks, performance profiling tools, and regression suites
• Advocate ML performance best practices, influencing framework and compiler enhancements across teams
• Provide technical leadership and mentorship to development teams implementing architectural solutions
• Ensure adherence to security, scalability, and maintainability standards in solution design

Languages