Senior Staff Machine Learning Engineer 2

At PowerSchool we are a dedicated team of innovators guided by our shared purpose of powering personalized education for students around the world. From the central office to the classroom to the home PowerSchool supports the entire educational ecosystem as the global leader of cloud-based software for K-12 education. Our employees make it all possible and a career with us means youre joining a successful team committed to engaging empowering and improving the K-12 education experience everywhere.

Team Overview

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.

This position under the general direction of the Lead and/or Manager Machine Learning Engineering will be responsible for technical and development support for our award-winning K-12 software. This role will help in all AI/Generative AI/Agents products in the areas of engineering data deployment and infrastructure.

• Uses Generative AI models (GPT4 Claude Gemini) other LLMs Agents and LangChains CrewAI Strands to build different AI driven smart solutions
• Experience in building GenAI based complex workflow in auto-pilot or co-pilot mode which run in production
• Experience in building Responsible AI guardrails and metrics around traditional AI or Gen AI models
• Familiar with latest Gen AI Agent based products and AI coding tools in the market
• Fundamental understanding of NLP transformer and embedding space
• Design and implement Machine learning models and data ingestion pipelines
• Experience in building software products used in production
• Create and maintain optimal data pipeline architecture by assembling large complex data sets to meet functional and non-functional business requirements
• Identify and implement internal process improvements including automating manual processes optimizing data delivery and redesigning infrastructure for greater scalability
• Support the building of machine learning data platforms and infrastructure required for optimal data extraction transformations and loading of data from a wide variety of data sources
• Work with architecture data and design teams to assist with data related technical issues and support data infrastructure needs
• Deploy ML models in AWS environment specifically in AWS Sage Maker environment
• Implement Model Monitoring Data Quality Checks Data Drifts in Inference Pipelines
• Support ML teams in the delivery of continuous integration continuous deployment providing templates and patterns
• Perform root cause analysis for production issues where the root cause is in infrastructure environment configuration or deployment routines; understand when to escalate to product development teams; remediate root causes and implement preventative actions
• Own the AWS stack which comprises all ML resources and collaborate on managing ML infrastructure costs
• Establish standards and practices around MLOps including governance compliance and data security
• Uses customer management system to provide status on open customer issues and properly verifies when an issue can be closed

• Qualifications include:
  • At least 5 years of experience within the full software development lifecycle from planning through deployment and maintenance
  • Youre a highly technical research engineer with a strong understanding of the latest advancements in AI especially GenAI and LLMs.
  • You have 5 years of professional experience in software engineering machine learning or applied research with a proven ability to drive high-impact AI initiatives end to end.
  • Strong working knowledge of deep learning machine learning and statistics
  • Experience related to AWS services such as SageMaker Bedrock EMR S3 OpenSearch Service Step Functions Lambda and EC2
  • Hands on experience with deep learning (e.g. CNN RNN LSTM Transformer) machine learning CV GNN or distributed training
  • Strong communication skills with attention to detail and ability to convey rigorous mathematical concepts and considerations to non-experts
  • Demonstrated ability to design implement and scale machine learning workflows (ML OPs); including deployment and delivery of production-ready model APIs
  • Proficiency with at least one machine learning lifecycle platform (Sagemaker MLFlow TensorFlow etc.) orchestration platform (Airflow Dagster etc.) and data platform like SnowFlake/DataBricks
  • You have a track record of implementing cutting-edge research into robust scalable and well-tested code.
  • You bring a strong engineering mindset and write clean efficient code that performs reliably in production.
  • Youve had some experience mentoring or guiding othersformally or informallyand are excited by the opportunity to help shape a new team.
  • You measure the impact of your work and provide clear visibility into what youre building and why it matters.
  • Experience working with ML engineers to build tooling and automation to support the entire ML engineering lifecycle from experimentation to production operations
  • Experience with Kubernetes and ML CI/CD workflows
  • 3 years experience with AWS or other public cloud platforms (GCP Azure etc.)
  • Excellent verbal and written communication skills.
  • Experience with Infrastructure-as-Code tools and frameworks
  • Bachelors degree in computer science data science mathematics or a related field. Masters degree preferred

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

#LI-NB1