đ 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.
If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.
Writer is on the lookout for a dynamic Principal Performance Engineer to join our innovative team in New York City. If you're passionate about pushing the boundaries of Generative AI technology, you're in the right place! In this unique role, you'll take charge of optimizing our cutting-edge Large Language Models (LLMs) and Retrieval Augmented Generation (RAG) systems for unmatched performance and reliability. You'll dive deep into identifying performance bottlenecks and optimizing resource usage to ensure our users enjoy a seamless experience. Collaborating closely with AI researchers, software engineers, and infrastructure teams, you will help shape the future of AI solutions while establishing performance benchmarks and providing mentorship to our engineering team. With more than 10 years of experience in performance engineering and a solid background in AI/ML technologies, you'll employ your skills in LLM capacity planning, infrastructure optimization, and advanced profiling techniques. Your role will also include evaluating new technologies, developing performance dashboards, and enhancing LLM inference capabilities. At Writer, we thrive on innovation and teamwork, and we believe that your expertise can help propel us to extraordinary heights! Get ready to lead us into the future of Generative AI!
Writer is the full-stack generative AI platform for enterprises. We empower your entire organization â support, operations, product, sales, HR, marketing, and more.
198 jobsSubscribe to Rise newsletter