5-7 years of experience in software engineering or related fields
Proficiency in Rust and/or C++ programming languages
Familiarity with deep learning frameworks such as PyTorch and/or JAX
Experience porting applications to specialized hardware
Solid knowledge of systems, including Linux and high-speed interconnects
Responsibilities
Support the adaptation of advanced models to our AI architecture
Scale and enhance the runtime for multi-node inference
Optimize communication layers using Sohu's collectives
Develop tools for performance analysis and debugging
Contribute to all aspects of the software stack
Benefits
Comprehensive medical, dental, and vision coverage
$2,000/month housing subsidy for nearby residents
Complimentary daily office meals
Relocation assistance for new hires moving to West San Jose
Full Job Description
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.
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.
You may be a good fit if you have
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.
Strong candidates may also have experience with (Nice-to-have qualifications)
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.
Benefits
Medical, dental, and vision packages with generous premium coverage
$500 per month credit for waiving medical benefits
Housing subsidy of $2.5k 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
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.
