Solution Architect (Kernel Optimization & ML Performance)

Join as a Solution Architect (Kernel Optimization & ML Performance) and shape scalable solutions that drive business value. Work at the intersection of hardware acceleration and machine learning infrastructure, translating performance and optimization requirements into robust architectural designs.

Collaborate with global, cross-functional teams, including ML researchers, compiler engineers, and systems architects. Influence the technical direction, quality, and efficiency of high-performance ML solutions for large-scale AI workloads.

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 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

Requirements

• Bachelor’s or Master’s degree in Computer Science or equivalent practical experience
• 12+ years of software engineering experience, including 5+ years in architecture or technical leadership roles
• In-depth knowledge of ML frameworks: JAX, PyTorch, and TensorFlow, and core ML concepts
• Hands-on expertise in kernel-level performance optimization targeting TPUs and GPUs
• Strong C++ and Python experience for high-performance computing
• Solid grasp of compiler principles, graph optimizations, and toolchains such as MLIR or OpenXLA
• Proven experience designing scalable, production-grade ML systems with an emphasis on performance and efficiency
• Excellent communication, stakeholder management, and solution delivery skills

Nice to have

• Familiarity with emerging hardware accelerators, heterogeneous compute, and scale-out performance patterns
• Experience with autotuning systems, benchmarking methodologies, and performance profiling tools
• Contributions to open-source projects in ML performance, kernels, or developer infrastructure
• Ability to present architectural solutions to technical and non-technical audiences

We offer

• Work with top technical talent driving innovation in AI, cloud, and digital platform modernization
• Supportive team and agile, startup-like culture
• Hybrid work model and opportunity to work remotely within Poland
• Opportunity to work abroad for up to 60 days annually
• Business-driven relocation opportunities
• Career development programs
• Thought leadership, mentoring, soft-skills, and well-being programs
• Certification opportunities: Anthropic, Gemini, GCP, Azure, and AWS
• English classes
• Stable pay
• Participation in the Employee Stock Purchase Plan with a 15% discount
• Benefits package, including health insurance, multisport, and shopping vouchers
• Referral bonuses up to $2,000
• Offices with entertainment and relaxation zones, table tennis, football, free snacks, coffee, and more
• Corporate, social, and well-being events

Benefits listed above are available to employees only. Contractor engagement is available; B2B cooperation terms are agreed individually. Only selected candidates will be contacted.