Etched
Inference Software Engineer
1y ago
175000 –275000 USD / yearUSAc++rustpytorchjaxlinux
Develop and optimize software for inference workloads on advanced hardware architecture, including model porting and runtime scaling.
Responsibilities
- Support porting state-of-the-art models to our architecture. Help build programming abstractions and testing capabilities to rapidly iterate on model porting.
- Build, enhance, and scale our runtime, including multi-node inference, intra-node execution, state management, and robust error handling.
- Optimize routing and communication layers using our collectives.
- Utilize performance profiling and debugging tools to identify bottlenecks and correctness issues.
Requirements
- Proficiency in C++ or Rust.
- Understanding of performance-sensitive or complex distributed software systems like Linux internals, accelerator architectures (e.g. GPUs, TPUs), Compilers, or high-speed interconnects (e.g. NVLink, InfiniBand).
- Familiarity with PyTorch or JAX.
- Ported applications to non-standard accelerator hardware or hardware platforms.
Nice to have
- Developed 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.
Conditions
- Medical, dental, and vision packages with generous premium coverage $500 per month credit for waiving medical benefits
- Housing subsidy of $2k per month for those living within walking distance of the office
- Relocation support for those moving to San Jose (Santana Row)
- Various wellness benefits covering fitness, mental health, and more
- Daily lunch + dinner in our office
- Unlimited compute budget subject to ROI justification
Other
- Etched is building hardware for frontier intelligence. We co-design chips, racks, software, and manufacturing to deliver best-in-class throughput and latency across both prefill and decode workloads. Our first products are heavily focused on inference . 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.
- Etched believes in the Bitter Lesson . We are the first inference-focused frontier AI system, betting early on transformer and transformer-like architectures and on increasing model sizes. Our addressable market is the entirety of inference, unlike many of our competitors.
- We are a fully in-person team in San Jose (Santana Row), 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 and work across disciplines as needed.