Hiring: Software Engineer II Company: Torentify Location: Seattle, WA Job Posted Time: 2026-09-17 00:13:26 Employment Type: Internship Target Skills & Keywords : AI, AWS, LLM, Machine Learning, PyTorch, Systems Design, TensorFlow, Transformers About the job Experience: •5+ years of programming experience. •5+ years of system design or architecture experience. •5+ years of full software development lifecycle experience. •Machine learning and large-scale LLM training experience with PyTorch is required. Required Skills: •Annapurna Labs (U.S.), part of AWS, is seeking a Senior Software Engineer specializing in AI/ML and distributed training for AWS Neuron. The role focuses on developing, enabling, and performance-tuning distributed training solutions for large-scale machine learning models running on AWS Trainium and Inferentia accelerators. •Lead efforts to build distributed training support into PyTorch, the AWS Neuron compiler, and runtime stacks. •Enable distributed training strategies and optimize ML models for peak performance and efficiency on AWS custom silicon, including Trainium servers. •Develop and tune distributed training solutions for large-scale machine learning models and LLMs. •Interface directly with distributed training strategies including FSDP (Fully-Sharded Data Parallel), Pipeline Parallelism (PP), and Context Parallelism. •Develop and enable distributed training capabilities using frameworks and libraries such as torchtitan, torchtune, HF RL, and DeepSeek. •Support machine learning model families including GPT-OSS, Quen, Llama, Stable Diffusion, and Vision Transformers (ViT). •Work closely with chip architects, compiler engineers, and runtime engineers. Qualifications: •Bachelor’s degree in Computer Science or equivalent. •5+ years of non-internship professional software development experience. •5+ years of programming experience with at least one software programming language. •5+ years of experience leading the design or architecture of new and existing systems, including design patterns, reliability, and scaling. •5+ years of experience across the full software development lifecycle, including coding standards, code reviews, source control management, build processes, testing, and operations. •Master’s degree in Computer Science or equivalent. •Software engineering expertise with PyTorch, JAX, or TensorFlow. •Bachelor’s degree in Computer Science or equivalent is required. •Master’s degree in Computer Science or equivalent is preferred. Compensation: •$168,100 - $227,400 / year •Flexible work environment (work from home / hybrid options) Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!