Job Details
Senior Computer Architect
Our client aims to revolutionize hardware and systems through a software-first approach, empowering AI innovators to surpass previous limits. The goal is to alleviate computational burdens, streamline model training and deployment, and ultimately maximize societal benefits from this transformative technology.
Their SPU (Spatial Processing Unit) stands as the pinnacle of programmable digital processors, significantly reducing AI-related costs. Efforts span across all engineering layers, encompassing hardware, runtime compilers, kernel optimization, algorithm development, and software architecture.
Looking for a Senior Computer Architect to drive optimization for state-of-the-art AI technology, enhancing code efficiency on our specialized hardware. You will join a collaborative team dedicated to innovative problem-solving and quality product creation. Make a lasting impact in AI's future.
Responsibilities:
- Lead the design, enhancement, and maintenance of a next-generation SPU compiler.
- Propose and implement enhancements to our Intermediate Representation (IR) to accommodate emerging trends in machine learning model architectures.
- Develop novel compiler passes and scheduling techniques to optimize code generation.
- Employ state-of-the-art parallelization and partitioning methodologies to automate kernel generation and exploit optimized kernels.
- Engage in rapid prototyping and data-driven exploration to evaluate new concepts.
- Benchmark and analyze outputs on SPU hardware, ensuring peak performance.
- Collaborate closely with hardware and software teams to align with the evolving requirements of ML engineers and drive architectural improvements.
- Develop tools for performance bottleneck analysis.
Qualifications:
- In-depth understanding of Computer Architecture, Microarchitecture, and Computer Science
- Expertise in designing and optimizing algorithms for AI workloads
- Extensive experience with simulation modeling
- Awareness of emerging trends and advancements in Computer Architecture
- Excellent problem-solving and analytical abilities
- Proficient in written and verbal communication
- Capable of working independently and collaboratively in a remote setting
- Ph.D. or Master's degree in Computer Science or a related discipline
- FPGA implementation experience