Lead AI Engineer

Job Title

PowerSchool is seeking a dedicated team of innovators guided by our shared purpose of powering personalized education for students around the world.

Job Description

Our Research & Development (R&D) team is the technical talent at the heart of our product suite, overseeing the product development lifecycle from concept to delivery. From engineering to quality assurance to data science, the R&D team ensures our customers seamlessly use our products and can depend on their consistency.

Responsibilities

Responsibilities include:

• Leading architecture and design for a new multi-tenant reporting platform built on AWS serverless technologies.
• Providing hands-on development using Python, TypeScript, SQL, and AWS services including Lambda, Step Functions, API Gateway and others.
• Developing AI-driven pipelines that extract, translate, and validate business logic and report parameter screens from the legacy platform.
• Defining coding standards, best practices, and development patterns across the engineering team.
• Mentoring senior and mid-level engineers; conducting design and code reviews.
• Partnering with Product, Data, QA, and Architecture teams to align technical and business requirements.
• Designing, developing, testing, deploying, and supporting AI software components including Agent Orchestration, large language model inference, similarity search, guardrails, model evaluation, experimentation, governance, and observability, etc.
• Leveraging a broad stack of Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, Nemo Guardrails, PyTorch, and more.
• Inventing and introducing state-of-the-art LLM optimization techniques to improve the performance — scalability, cost, latency, throughput — of large scale production AI systems.
• Ensuring platform scalability, security, tenant isolation, and operational excellence.

Qualifications

Minimum qualifications include:

• Extensive experience with AWS serverless architectures.
• Expert-level proficiency in Python and TypeScript.
• Strong SQL background; Snowflake experience is a plus.
• Demonstrated experience building or integrating AI-driven systems (LLMs, embeddings, AI agents, transformation pipelines).
• Solid experience with distributed systems, data pipelines, or modernizing legacy platforms.
• Proven leadership in technical strategy and mentoring.
• Experience designing or operating multi-tenant platforms.
• Strong knowledge of ETL/ELT architecture patterns.
• Prior experience with AI for code generation, metadata extraction, or workflow automation.
• Strong documentation, communication, and stakeholder-management skills.

Preferred qualifications include:

• Experience developing AI and ML algorithms or technologies (e.g. LLM Inference, Similarity Search, Agents and VectorDBs, Guardrails, Memory) using Python, Typescript, React.
• Experience developing and applying state-of-the-art techniques for optimizing training and inference software to improve hardware utilization, latency, throughput, and cost.
• Passion for staying abreast of the latest AI research and AI systems, and judiciously apply novel techniques in production.

Compensation & Benefits

PowerSchool offers the following benefits:

• Comprehensive Insurance Coverage (including Medical, Dental, Vision, Pharmacy benefits, Life Insurance and AD&D)
• Flexible Spending Accounts and Health Savings Accounts
• Short-Term Disability and Long-Term Disability
• Comprehensive 401(k) plan
• Generous Parental Leave
• Unrestricted paid time off (known as Discretionary Time Off - DTO)
• Wellness Program, including ClassPass & Employee Assistance Program
• Tuition Reimbursement
• Optional Benefits: Pet Insurance, Identity Theft Protection, Student Debt Repayment Program and Prepaid Legal coverage

A reasonable estimate of the base compensation range for this position is $92,800 - $170,300 USD. This compensation range is specific to the United States and it incorporates many factors including but not limited to an applicant's skills and prior relevant experience and training; licensures, degrees, and certifications; internal equity; internal pay ranges; and market data/range parameters.

EEO Commitment

PowerSchool is committed to a diverse and inclusive workplace. PowerSchool is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status. Our inclusive culture empowers PowerSchoolers to deliver the best results for our customers. We not only celebrate the diversity of our workforce, we celebrate the diverse ways we work. If you have a disability and need an accommodation regarding our recruiting process, please let us know by emailing View email address on click.appcast.io.