SIGN IN
Machine Learning Compiler Architect jobs in United States
cer-icon
Apply on Employer Site
company-logo

CyberX Info System · 1 day ago

Machine Learning Compiler Architect

CyberX Info System is seeking a Machine Learning Compiler Architect to design and develop a robust compiler architecture that interacts effectively with their NPU. The role involves implementing advanced graph optimizations, developing algorithms for memory management, and ensuring seamless integration with hardware to maximize performance for AI workloads.
Crypto & Web3Art & CreativeSoftwareAppsMarketingInformation TechnologyBlockchainDigital MarketingGraphic DesignMobile AppsSEOWeb DesignWeb Development
Hiring Manager
Irfan Ahmad
linkedin

Responsibilities

Design and develop a robust compiler architecture that effectively interacts with our NPU
Implement advanced graph optimizations that incorporate both hardware agnostic and hardware specific enhancements
Develop and optimize algorithms for tiling and memory management to efficiently utilize the NPU's resources
Create sophisticated optimization passes for neural network inference and training workloads
Map high-level operations to optimized library macros and convert them into hardware-level instructions
Generate and manage DMA commands to facilitate data movement and operation within the hardware ecosystem
Collaborate with hardware engineers and system architects to ensure seamless integration and maximal performance of the NPU
Implement efficient scheduling and resource allocation algorithms for concurrent AI workload execution
Stay updated with the latest trends and advancements in compiler technology and machine learning to continuously improve the compiler design
Lead research initiatives in advanced compilation techniques for AI accelerators
Drive adoption of cutting-edge optimization strategies and compilation methodologies
Mentor engineering teams on compiler design principles and best practices

Qualification

Compiler developmentMachine learning algorithmsTVMIREEXLAMLIRLLVMC++PythonGraph optimization techniquesMemory management strategiesHardware architecture principlesExpert communicatorTeam motivationStakeholder managementCross-functional project managementTechnical leadershipStrategic thinkingCollaboration across teams

Required

Expert communicator across cultural and team boundaries
Expertise in motivating teams and fostering a collaborative and productive environment
Background in managing multiple and competing stakeholder interests; establishing trust, clear roles and responsibilities, and goodwill between partner engineering organizations
Experience managing cross-functional and/or cross-team projects
Technical leadership experience with ability to mentor engineering teams
Strategic thinking capabilities with focus on long-term architectural decisions
Collaborate and work with multiple teams across geographies and time zones
12+ years of experience in compiler development or architecture, particularly targeting AI or ML hardware accelerators
Strong understanding of machine learning algorithms and their computational implications
Working experience with TVM, IREE, XLA, MLIR or LLVM
Proficiency in programming languages such as C++ and Python
Experience with graph optimization techniques and memory management strategies in compilers
Demonstrated ability to translate high-level functional requirements into detailed technical designs
Deep knowledge of hardware architecture principles and AI accelerator design concepts
Proven track record of leading compiler architecture projects from concept to production deployment

Company

CyberX Info System

twittertwittertwitter
company-logo
Cyberx Info System's unique brand differentiation is our relentless focus on customer-centric mobile app development.

Funding

Current Stage
Growth Stage

Leadership Team

leader-logo
Rachel Noor
Company Owner
linkedin
Company data provided by crunchbase