E

Etched

Designs chips and servers for running AI models

Inference Software Engineer

Applied Machine LearningUpdated on Nov 8
Full-timeSan Jose, CA, USA$150k - $400k /yr

About the Job

Requirements

  • Proficiency in C++ or Rust.
  • Understanding of performance-sensitive or complex distributed software systems, such as Linux internals, accelerator architectures (for example, GPUs or TPUs), compilers, or high-speed interconnects (for example, NVLink or InfiniBand).
  • Familiarity with PyTorch or JAX.
  • Experience porting applications to non-standard accelerator hardware or hardware platforms.

Responsibilities

  • Support porting state-of-the-art models to the company’s architecture and help build programming abstractions and testing capabilities to rapidly iterate on model porting.
  • Build, enhance, and scale the runtime, including multi-node inference, intra-node execution, state management, and robust error handling.
  • Optimize routing and communication layers using collectives.
  • Use performance profiling and debugging tools to identify bottlenecks and correctness issues.

Desired Qualifications

  • Experience developing 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 understanding of Transformer architectures, particularly Mixture-of-Experts (MoE).
  • Experience building applications with extensive SIMD (Single Instruction, Multiple Data) optimizations for performance-critical paths.

About the Company

Etched

Designs chips and servers for running AI models

Etched designs AI chips and the server racks built around them for AI companies and cloud providers, focusing only on inference, the work of running trained models to answer requests. In 2024, it unveiled a chip hard-wired to run only transformers, the type of AI model behind ChatGPT, after betting at its founding that they would dominate AI. Etched develops its chips, racks, and software together, and each rack links many chips through a shared pool of fast memory so very large models with long inputs can answer quickly. It earns money by selling these complete systems rather than individual chips. Etched was founded in 2022 and is based in San Jose, California.

Company size
201-500
Company stage
Series D
Total funding
$1.6B
Headquarters
San Jose, CA
Founded
2022

Recently Posted Jobs

Explore more opportunities at Etched

See more jobs