Job description
📐 About this role
Writer is seeking a highly skilled and motivated Principal Performance Engineer to lead the performance optimization of our cutting-edge Generative AI technology stack. This role is critical in ensuring the scalability, efficiency, and reliability of our Large Language Models (LLMs) and Retrieval Augmented Generation (RAG) systems. You will be a key driver in identifying and resolving performance bottlenecks, optimizing resource utilization, and ensuring a seamless user experience. You will work closely with our AI research, software engineering, and infrastructure teams to deliver world-class AI solutions.
🦸🏻♀️ Your responsibilities
Performance Leadership:
Define and implement performance engineering strategies for our Generative AI full stack, including services, application, LLMs, RAG pipelines, and related infrastructure.
Lead performance testing, profiling, and analysis efforts to identify and resolve performance bottlenecks.
Establish and maintain performance benchmarks and SLAs for critical AI services.
Provide technical leadership and mentorship to performance engineering team members.
LLM Capacity and Tuning:
Analyze and improve LLM inference performance, including latency, throughput, and resource utilization.
Develop and implement strategies for LLM capacity planning and scaling.
Collaborate with AI researchers to evaluate and improve LLM model architectures and training techniques for performance.
Optimize LLM inference through techniques such as quantization, distillation, and optimized kernel implementation.
RAG Performance Optimization:
Design and implement performance tests for RAG pipelines, including retrieval, ranking, and generation components.
Identify and optimize performance bottlenecks in RAG systems, such as database queries, vector search, and document processing.
Evaluate and optimize RAG system architectures for scalability and efficiency.
Tune vector databases for optimal recall and latency.
Infrastructure Optimization:
Collaborate with infrastructure teams to optimize hardware and software configurations for AI workloads.
Evaluate and recommend new technologies and tools for performance monitoring and analysis.
Develop and maintain performance dashboards and reports to track key metrics.
Optimize GPU utilization and memory management for LLM inference.
Collaboration and Communication:
Work closely with AI researchers, software engineers, and product managers to ensure performance requirements are met.
Communicate performance findings and recommendations to stakeholders at all levels.
Stay up-to-date with the latest developments in Generative AI and performance engineering.
⭐️ Is this you?
Education:
- Bachelor’s degree in Computer Science, Engineering, or a related field (Master’s preferred).
Experience:
10+ years of experience in performance engineering, with a focus on large-scale distributed systems.
2+ years of experience working with AI/ML technologies
Proven experience in performance testing, profiling, and analysis of complex software systems.
Deep understanding of NLP architectures, training, and inference.
Experience with vector databases and search technologies.
Experience with cloud computing platforms (e.g., AWS, Azure, GCP) and containerization technologies (e.g., Docker, Kubernetes).
Strong programming skills in python.
Experience with performance analysis tools (e.g., profilers, debuggers, monitoring tools).
Skills:
Strong analytical and problem-solving skills.
Excellent communication and collaboration skills.
Ability to work in a fast-paced and dynamic environment.
Passion for AI and a desire to push the boundaries of performance engineering
#LI-Remote
🍩 Benefits & perks (US Full-time employees)
Generous PTO, plus company holidays
Medical, dental, and vision coverage for you and your family
Paid parental leave for all parents (12 weeks)
Fertility and family planning support
Early-detection cancer testing through Galleri
Flexible spending account and dependent FSA options
Health savings account for eligible plans with company contribution
Annual work-life stipends for:
Home office setup, cell phone, internet
Wellness stipend for gym, massage/chiropractor, personal training, etc.
Learning and development stipend
Company-wide off-sites and team off-sites
Competitive compensation, company stock options and 401k
Writer is an equal-opportunity employer and is committed to diversity. We don’t make hiring or employment decisions based on race, color, religion, creed, gender, national origin, age, disability, veteran status, marital status, pregnancy, sex, gender expression or identity, sexual orientation, citizenship, or any other basis protected by applicable local, state or federal law. Under the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
By submitting your application on the application page, you acknowledge and agree to Writer’s Global Candidate Privacy Notice.
