Logo for Lexia Learning

Machine Learning Software Engineer II

Role overview

Qualifications

  • 2–5 years of industry experience in ML engineering, software engineering, or data science with a proven track record of deploying models to production
  • Deep hands-on experience with AWS ECS and AWS Lambda, Docker containerization, and event-driven architectures
  • Strong proficiency in Python, C++, and/or Java, with experience following industry-standard coding practices
  • Hands-on experience with PyTorch and ML libraries (Scikit-Learn, TensorFlow); knowledge of model optimization (PyTorch to ONNX, ONNX Runtime) and memory-efficient attention (e.g., Flash Attention)

Responsibilities

  • Full-Lifecycle ML Development: Lead the transition of machine learning models from theoretical prototypes into scalable, high-performance production systems
  • AWS Cloud Architecture Deployment: Architect and deploy ML solutions utilizing AWS ECS for containerized workloads and AWS Lambda for serverless, event-driven inference pipelines
  • Model Inference Optimization: Optimize PyTorch models for production deployment by converting to ONNX formats and applying quantization, pruning, ONNX Runtime
  • Infrastructure Engineering Best Practices: Establish reliable CI/CD pipelines and robust, secure, and reproducible deployments across the AWS ecosystem

About the company

Lexia Learning logo

Lexia Learning

E-Learning / EdTech

Lexia believes that strong literacy foundations have impact far beyond the classroom—that they are essential to our ability to communicate, express ourselves, and engage with the world around us. We believe everyone has a right to literacy, and that as educators, we have an obligation to deliver. We believe every student deserves to be taught literacy in a way that works for them, no matter who they are, where they live, what level they are at, and what learning challenges they face. We believe every teacher deserves the support they need to teach literacy confidently and effectively. We know that literacy is fully within reach for the vast majority of learners (both students and teachers) when instruction is based on the science of reading—with sound methodology; with engaging experiences supported by innovative, committed experts; and with a focus on the value of each individual learner. We’re Lexia. We’re all for literacy because literacy can and should be for all.

Company details

IndustryE-Learning / EdTech
Company size501 - 1000

Your match analysis

See how your profile stacks up against this role.

We compared the job requirements to your profile to show where you're strong and where you fall short.

Job description

Cambium Learning® Group is an award-winning educational technology solutions leader dedicated to helping all students reach their potential through individualized and differentiated instruction. Using a research-based, personalized approach, Cambium Learning Group delivers SaaS resources and instructional products that engage students and support teachers in fun, positive, safe and scalable environments. These solutions are provided through Learning A-Z® (online differentiated instruction for elementary school reading, writing and science), ExploreLearning® (online interactive math and science simulations, a math fact fluency solution, and a K–2 science solution), Voyager Sopris Learning® (blended solutions that accelerate struggling learners to achieve in literacy and math and professional development for teachers), and VKidz Learning (online comprehensive homeschool education and programs for literacy and science). We believe that every student has unlimited potential, that teachers matter, and that data, instruction, and practice are the keys to success in the classroom and beyond.

Job Overview:

We are seeking a talented Machine Learning Engineer II to join our CAI machine learning and scoring development team. In this role, you will be the crucial bridge between applied research and production systems. Working alongside a cross‑functional group of mathematicians, computer scientists, psychometricians, and statisticians, you will design and deploy custom machine learning solutions for our clients and internal platforms.

The ideal candidate is a full‑stack ML practitioner who is equally comfortable discussing algorithmic design with researchers and architecting scalable, low‑latency production systems. You will own the full software development lifecycle—transforming research prototypes into optimized, production‑ready solutions using modern AWS infrastructure such as SageMaker, ECS, and Lambda, with an emphasis on high‑throughput inference and PyTorch‑to‑ONNX model optimization.

Job Responsibilities:

  • Full-Lifecycle ML Development: Lead the transition of machine learning models from theoretical prototypes into scalable, high-performance production systems.
  • AWS Cloud Architecture & Deployment: Architect and deploy ML solutions utilizing AWS ECS (Elastic Container Service) for containerized workloads and AWS Lambda for serverless, event-driven inference pipelines.
  • Model & Inference Optimization: Optimize PyTorch models for production deployment by converting them to ONNX formats. Apply advanced inference optimization techniques (quantization, pruning, ONNX Runtime) and memory-efficient attention mechanisms like Flash Attention to minimize latency and maximize throughput.
  • Infrastructure & Engineering Best Practices: Champion infrastructure best practices for machine learning systems, establishing reliable CI/CD pipelines, and ensuring robust, secure, and reproducible deployments across the AWS ecosystem.
  • Algorithm Engineering: Design, develop, and evaluate algorithms that generate descriptive, diagnostic, predictive, and prescriptive insights from both structured and unstructured data.
  • Robust Software Engineering: Write clean, efficient, and well-tested code. Complete rigorous testing, debugging, and documentation to ensure seamless installation and long-term maintenance.
  • Cross-Functional Collaboration: Actively participate in research discussions, requirements gathering, and system design alongside domain experts to build tailored scoring and ML solutions.

Job Requirements:

  • Experience: 2–5 years of industry experience in Machine Learning Engineering, Software Engineering, or Data Science, with a proven track record of architecting and deploying models to production.
  • Cloud & MLOps Infrastructure: Deep, hands-on experience with the AWS ecosystem, specifically AWS ECS and Lambda. Solid understanding of containerization (Docker) and event-driven architectures.
  • Programming Proficiency: Strong proficiency in modern programming languages used in ML (e.g., Python, C++, Java) and familiarity with industry-standard coding practices.
  • ML Frameworks & Advanced Optimization: Hands-on experience with PyTorch and other machine learning libraries (e.g., Scikit-Learn, TensorFlow). Deep understanding of model optimization pipelines, including PyTorch to ONNX conversions, ONNX Runtime, and scaling attention mechanisms (e.g., Flash Attention).
  • Data Systems: Experience working with large-scale computing frameworks, data analysis systems, and relational/non-relational databases.

Nice to Have's:

  • AWS SageMaker: Experience utilizing AWS SageMaker for managed model training and hosting.
  • Advanced LLMOps & Fine-Tuning: Hands-on experience applying modern parameter-efficient fine-tuning methods (such as LoRA and qLoRA) to large language models.
  • AI Agents: Experience building, integrating, and deploying autonomous or semi-autonomous AI agents to automate complex workflows and connect ML models with external tools/APIs.
  • NLP Expertise: Proven experience and familiarity with deep learning technologies applied specifically to Natural Language Processing (NLP) and complex text-based modeling.
  • Cross-Disciplinary Collaboration: Experience collaborating with specialized researchers (e.g., psychometricians, statisticians) to operationalize complex mathematical concepts.
  • Infrastructure as Code: Experience implementing IaC using tools like Terraform or AWS CloudFormation.
  • Model Monitoring: Experience setting up comprehensive model monitoring systems to detect data drift, concept drift, and model degradation in production AWS environments.

To apply for this opportunity, simply click on the “Apply” button and submit a cover letter and resume.

An Equal Opportunity Employer

We are dedicated to fostering a culture that celebrates unique backgrounds, ideas, and experiences. All qualified applicants will receive consideration for employment without discrimination on the basis of race, color, religion, sex, gender, gender identity/expression, sexual orientation, national origin, protected veteran status, or disability.

Apply once. Then go straight to the hiring manager.

After you apply, unlock the direct contact details of the people who actually make the call. A quick follow-up makes you 5x more likely to land an interview.

MR

Marcus Rivera

Chief Revenue Officer

m.rivera@company.com
linkedin.com/in/marcusrivera
Unlocked after you apply
·

Machine Learning Engineer Related jobs

Other jobs at Lexia Learning

Premium

Reach out to the hiring manager directly.

Gain access to the contact details of the hiring managers who actually decide, and reach out to network with them directly. That, plus more when you upgrade:

  • Full match report with fit score and gaps
  • Career diagnostics on how recruiters read you
  • Curated company matches and warm intros
  • 48h early access to new roles

Cancel anytime.