Senior ML Engineer, Justworks

• 5+ years of experience in Machine Learning, Data Science, or related fields
• Strong proficiency in Python, SQL, and ML frameworks (TensorFlow, PyTorch, Scikit-Learn)
• Experience with building, training, and optimizing ML models (regression, classification, recommendation, NLP, etc.)
• Familiarity with MLOps best practices (model deployment, versioning, monitoring)
• Experience with cloud platforms (AWS, GCP, Azure) and distributed computing
• Exposure to vector databases, embeddings, and model fine-tuning is a plus
• Knowledge of deep learning architectures (CNNs, RNNs, Transformers) is a strong advantage

Desirable

• We are looking for a Senior Machine Learning Engineer to build and optimize machine learning systems that power our products. This role offers a strong career growth trajectory into AI Engineering, making it ideal for candidates who have deep ML expertise and are eager to work on cutting-edge AI technologies such as LLMs, generative AI, and scalable AI infrastructure
• As a Senior ML Engineer, you will develop and deploy machine learning models at scale, optimize model performance, and collaborate with cross-functional teams to drive impactful AI-driven solutions. Over time, you will gain exposure to AI research, deep learning frameworks, and next-gen AI architectures, with the opportunity to transition into an AI Engineering role
• Within the Justworks Data Services team, you’ll find a highly-collaborative, diverse group of individuals motivated to empower Justworkers to work fearlessly and responsibly with data. We are committed to growing our impact in the best interest of our customers and evolving how we work together in the best interest of our people & company. Come build the future with us!
• Machine Learning Engineering (Primary)
• Design, build, and deploy scalable ML models for production systems
• Work on feature engineering, model training, tuning, evaluation, and deployment
• Develop and maintain high-performance ML pipelines for data processing and inference
• Optimize ML models for latency, efficiency, and cost-effectiveness
• Implement robust monitoring and observability frameworks for ML models in production
• Work closely with data engineers, MLOps engineers, and product teams to integrate ML
• Solutions into business workflows
• AI Engineering Transition Pathway
• Gain hands-on experience in deep learning, LLMs, multimodal AI, and generative AI
• Learn and apply cutting-edge AI frameworks (e.g., PyTorch, TensorFlow, JAX, LangChain)
• Experiment with transformers, diffusion models, and reinforcement learning
• Contribute to the development of AI-powered applications and intelligent automation systems
• Engage in research-driven AI projects, attending relevant AI/ML conferences and workshops
• Collaborate with AI scientists and researchers to productionize advanced AI models
• As a Senior ML Engineer, how results are achieved is paramount for your success and ultimately result in our success as an organization. In this role, your foundational knowledge, skills, abilities and personal attributes are anchored in the following competencies:
• Good judgment - the exercise of critical thinking, analyzing and assessing problems and implications, identifying patterns, making connections of underlying issues, understanding risks and developing mitigation strategies, and taking ownership of the outcome
• Teamwork and communication - putting our collective best together through documentation, collaboration, relationship-building, listening, empathy, recruiting, and evangelism
• Influence and leadership - fostering a community of knowledge-sharing, collaboration, mentorship, and forward-thinking
• Skills and knowledge - the capacity to actively learn and apply specific domain knowledge, know-how, and best practices to continually enhance and improve
• In addition, all Justworkers focus on aligning their behaviors to our core values known as COGIS. It stands for:
• Camaraderie - Day to day you can be seen working together toward a higher purpose. You like to have fun. You’re an active listener, treat people respectfully, and have a strong desire to know and help others
• Openness - Your default is to be open. You're willing to share information, understand other perspectives, and consider new possibilities. You’re curious, ask open questions, and are receptive to thoughts and feedback from others
• Grit - You demonstrate grit by having the courage to commit and persevere. You’re committed, earnest, and dive in to get the job done well with a positive attitude
• Integrity - Simply put, do what you say and say what you'll do. You’re honest and forthright, have a strong moral compass, and strive to match your words with your actions while leading by example
• Simplicity - Be like Einstein: “Everything should be made as simple as possible, but no simpler.”
• Technologies Used
• Languages: SQL, Python, Terraform
• Stack: Snowflake, Airflow, Fivetran, dbt, ,GitHub, CircleCI, Tableau, Redash, Jira