NPU Architect: FPGA/ASIC Hardware Accelerators

TikTok Inc.; San Jose (CA); On-site; USD 162,000 - 317,000; Full time

Job summary

TikTok is seeking strong video codec design engineers to build hardware accelerators for advanced video encoding and processing. You will work on architecture design, performance analysis, and optimization for next-generation chips, focusing on NPU modules and DL inference performance.

You will model throughput and memory, define chip specifications, and drive HW-SW co-optimization with cross-disciplinary teams. Experience with SoC subsystems and FPGA emulation is valued.

Qualifications

• Master's degree or above in Microelectronics, Electronic Engineering, Integrated Circuits, Computer Science, or a related field.
• Familiarity with Tensor Core; relevant experience in hardware acceleration for large language models.
• Solid knowledge of computer architecture and digital integrated circuit design.
• Familiarity with deep learning algorithms in areas such as large language models and CNNs.
• Familiarity with distributed frameworks for large language model inference; understanding of how to compute the performance of AI model inference on chips.
• Strong independent problem-solving skills, good communication and teamwork abilities, a strong desire to learn, the initiative to stay current with evolving trends in AI algorithms and chip architecture, and a forward-looking technical vision.

Responsibilities

• Design hardware architecture, perform architectural performance analysis, and optimize key NPU modules in next-gen chips.
• Model and analyze performance of DL algorithms on chip systems; specify compute throughput, bandwidth, and memory.
• Optimize architecture for real-world app scenarios, enabling HW-SW co-optimization.
• Write architecture specifications and guide design, optimization, and verification of microarchitecture.

Skills

Tensor Core familiarity

DL hardware acceleration

Computer architecture

DL model inference on chips

Teamwork and communication

Problem solving

Education

Master's degree or above in Microelectronics/ Electronic Eng/ ICs/ CS

Tools

Palladium

ZeBu

FPGA prototyping

JasperGold

Job description

TikTok is seeking strong video codec design engineers to build hardware accelerators for advanced video encoding and processing. You will work on architecture design, performance analysis, and optimization for next-generation chips, focusing on NPU modules and DL inference performance.

You will model throughput and memory, define chip specifications, and drive HW-SW co-optimization with cross-disciplinary teams. Experience with SoC subsystems and FPGA emulation is valued.