Etched
Etched
Posted 3 months ago
Internship
San Jose, California
In Person
Smart Summary
Responsibilities
Contribute to the design and optimization of next-generation AI accelerators for transformer workloads. Responsibilities include porting models, scaling the runtime, and co-designing hardware instructions to maximize performance.
Qualifications
You are pursuing a Bachelor's, Master's, or PhD in a related field and possess proficiency in Python and C++. You should have an understanding of performance-sensitive distributed systems or accelerator architectures, and experience porting applications to non-standard hardware. Deep knowledge of transformer models or inference serving stacks is also highly desirable.
Must Have Skills for ATS
Python
C++
Linux internals
accelerator architectures
Compilers
high-speed interconnects
transformer model architectures
inference serving stacks
Rust
networking stacks
distributed systems
consensus protocols
consistency models
communication patterns
Mixture-of-Experts (MoE)
SIMD
PyTorch
JAX
Job Description
About Etched
Etched is building the world’s first AI inference system purpose-built for transformers - delivering over 10x higher performance and dramatically lower cost and latency than a B200. With Etched ASICs, you can build products that would be impossible with GPUs, like real-time video generation models and extremely deep & parallel chain-of-thought reasoning agents. Backed by hundreds of millions from top-tier investors and staffed by leading engineers, Etched is redefining the infrastructure layer for the fastest growing industry in history.
Job Summary
We are seeking talented fall or winter Architecture interns to join our team and contribute to the design of next-generation AI accelerators. This role focuses on developing and optimizing compute architectures that deliver exceptional performance and efficiency for transformer workloads. You will work on cutting-edge architectural problems and performance modeling over the course of your internship.
Key responsibilities
Support porting state-of-the-art models to our architecture. Help build programming abstractions and testing capabilities to rapidly iterate on model porting.
Assist in building, enhancing, and scaling Sohu’s runtime, including multi-node inference, intra-node execution, state management, and robust error handling.
Contribute to optimizing routing and communication layers using Sohu’s collectives.
Utilize performance profiling and debugging tools to identify bottlenecks and correctness issues.
Develop and leverage a deep understanding of Sohu to co-design both HW instructions and model architecture operations to maximize model performance
Implement high-performance software components for the Model Toolkit
You may be a good fit if you have
Progress towards a Bachelor’s, Master’s, or PhD degree in computer science, computer engineering, applied mathematics, or a related field
Proficiency in Python, C++
Understanding of performance-sensitive or complex distributed software systems, e.g. Linux internals, accelerator architectures (e.g. GPUs, TPUs), Compilers, or high-speed interconnects (e.g. NVLink, InfiniBand).
Ported applications to non-standard accelerator hardware or hardware platforms.
Deep knowledge of transformer model architectures and/or inference serving stacks (vLLM, SGLang, etc.)
Strong candidates may have some experience with
Proficiency in Rust
Low-latency, high-performance applications using both kernel-level and user-space networking stacks.
Deep understanding of distributed systems concepts, algorithms, and challenges, including consensus protocols, consistency models, and communication patterns.
Solid grasp of Transformer architectures, particularly Mixture-of-Experts (MoE).
Built applications with extensive SIMD (Single Instruction, Multiple Data) optimizations for performance-critical paths.
Familiarity with PyTorch or JAX.
Math competitions (AIME, AMC, etc)
We encourage you to apply even if you do not believe you meet every qualification.
Program details
12-week paid internship (fall or winter)
Generous housing support for those relocating
Daily lunch and dinner in our office
Based at our office in San Jose, CA
Direct mentorship from industry leaders and world-class engineers
Opportunity to work on one of the most important problems of our time
For any questions, contact internships@etched.com.
How we’re different
Etched believes in the Bitter Lesson. We think most of the progress in the AI field has come from using more FLOPs to train and run models, and the best way to get more FLOPs is to build model-specific hardware. Larger and larger training runs encourage companies to consolidate around fewer model architectures, which creates a market for single-model ASICs.
We are a fully in-person team in West San Jose, and greatly value engineering skills. We do not have boundaries between engineering and research, and we expect all of our technical staff to contribute to both as needed.
With Runway
See Your Fit for This Role
1-5 min
Your Score
?
Top Applicants
90%
Your Job Search Advantage
Key Gaps & Next Steps:
Address these in your resume & Interview
Top Strengths For This Role
Highlight these in your cover letter & interview
Your Interview Guide
A Personalized Interview Strategy
Freshest Opportunities
Never Miss a Good Fit
Get notified when jobs mach your criteria