AI Software Engineer

About Elastix AI We are building the next-gen AI inference platform. Description Job Title: Software Engineer, AI Inference Platform Company: ElastixAI, Inc. Location: Seattle, WA (Hybrid - 3 days/week in office) About ElastixAI ElastixAI is an early-stage startup building the next-generation AI inference infrastructure — co-designed across ML software and custom accelerator hardware . Our platform dynamically optimizes inference efficiency and scalability across diverse deployments, enabling adaptive, high-performance AI serving. Role Summary We’re looking for a systems-minded AI Software Engineer to join our core inference platform team. You’ll design and extend the low-level serving stack — hacking open-source frameworks like vLLM, SGLang, and TensorRT-LLM , building new model sharding and scheduling logic, and integrating deeply with our proprietary AI accelerator . This role sits at the intersection of ML systems, compiler/runtime engineering, and hardware-software co-design . Key Responsibilities Architect, extend, and optimize core components of our AI serving platform for throughput, latency, and scalability. Customize open-source serving frameworks (e.g., vLLM) for proprietary model ingestion and accelerator integration. Develop efficient model partitioning, scheduling, and memory management strategies for multi-device inference. Collaborate with ML engineers on model export and runtime optimization (quantization, graph transforms). Work closely with hardware engineers to influence accelerator interface design and performance tuning. Build APIs and runtime tools enabling flexible, PyTorch-native model deployment on our infrastructure. Profile, debug, and optimize across the full stack — from Python orchestration to C++ kernels and PCIe drivers. Required Qualifications BS/MS/PhD in Computer Science, Electrical/Computer Engineering, or related field. 3+ years of professional experience in systems programming, ML infrastructure, or distributed inference . Proficient in C++ and Python , with strong debugging and performance analysis skills. Deep familiarity with one or more LLM serving frameworks (vLLM, SGLang, TensorRT-LLM, DeepSpeed-Inference, etc.). Understanding of model deployment internals — token scheduling, KV caching, batching, and pipelined inference. Comfortable working close to the hardware abstraction layer — CUDA, PCIe, memory management, or runtime scheduling. Strong collaboration and communication skills; ability to work cross-functionally in a fast-paced startup environment. Preferred / Bonus Experience with hardware-aware ML optimization , compiler/runtime integration, or accelerator SDKs. Hands-on experience profiling GPU/accelerator workloads . Familiarity with containerized deployments (Docker/Kubernetes) . Exposure to distributed systems or large-scale inference clusters. Contributions to open-source ML or serving frameworks. What We Offer: A chance to be a foundational engineer in an innovative AI startup A dynamic and collaborative work environment and the change to have a significant impact on new technology The opportunity to work on challenging problems at the intersection of ML, software, and systems. Competitive compensation and startup equity package Comprehensive medical, dental, and vision coverage (100% paid by employer) Life insurance and AD&D Flexible Time Off (FTO) 12-paid holidays Paid parental leave Gym or fitness benefit Commuter benefit Weekly catered lunches in the office Investment in employee learning & development