Senior Software Engineer, Applied AI

About the role

Justworks is hiring a Senior Software Engineer focused on Applied AI to join their fully remote team. This role centers on building and improving artificial intelligence features that power the company's human resources and benefits platform. The engineer will work closely with product and data teams to deliver machine learning solutions that enhance the day to day experience for Justworks customers across multiple product areas. The successful candidate will take ownership of AI features from ideation through production, ensuring that models are reliable, fair, and aligned with customer needs and business objectives. The role reports to the engineering manager and collaborates with a team of data scientists and software engineers dedicated to advancing AI capabilities within the platform.

Key facts

What you'll do

• Design and implement machine learning models that solve real problems for Justworks users
• Collaborate with cross functional teams to define AI requirements and success metrics
• Build data pipelines and feature stores to support model training and inference
• Evaluate model performance and iterate on approaches to improve accuracy and reliability
• Write clean, maintainable code for production AI systems that operate at scale
• Partner with product managers to translate business needs into technical specifications
• Conduct experiments and analyze results to guide the direction of applied AI work
• Optimize existing models for latency, cost, and scalability in production environments
• Mentor junior engineers and contribute to the technical culture of the team
• Stay current with advances in AI research and assess their applicability to Justworks products
• Document models and findings to ensure knowledge sharing and reproducibility across the organization
• Work with data engineering teams to ensure data quality and integrity for AI workloads

Requirements

• Demonstrated experience building and deploying machine learning models in production settings at scale
• Strong programming skills in Python or a comparable language for data intensive engineering work
• Familiarity with common AI frameworks and libraries used for model development and training
• Understanding of software engineering principles including testing, version control, code review, and documentation
• Ability to work independently and communicate technical ideas clearly to diverse stakeholders and partners
• Experience working with structured and unstructured data to derive actionable insights and recommendations
• A bachelor's degree in computer science or a related technical field is required
• Comfort with remote collaboration and asynchronous communication across multiple time zones and regions
• Familiarity with REST APIs and building services that serve model predictions to end users
• Experience with monitoring and logging infrastructure to track model behavior and data drift in production

Nice to have

• Experience with large language models or generative AI techniques applied to business products and workflows
• Background in human resources technology or workplace benefits platforms and related domains
• Familiarity with cloud based machine learning services and model deployment tooling for production
• Contributions to open source projects or published work in AI related research fields
• Prior experience working at a fast growing startup or a company with a distributed workforce
• Knowledge of compliance and privacy regulations relevant to handling employee data in AI systems

Skills & tools

• Python for data processing and model development
• Machine learning frameworks such as TensorFlow or PyTorch
• SQL and data warehouse tools for working with large datasets
• Cloud platforms like AWS or GCP for model hosting and scaling
• Version control with Git and collaborative development workflows
• Containerization and orchestration tools for deploying AI services
• Docker and Kubernetes for containerizing and managing AI model deployments
• Experiment tracking tools such as MLflow or Weights and Biases for managing model iterations

Practical notes

• This is a remote position open to candidates located anywhere in the world without a fixed office requirement
• Justworks operates on a remote first basis with asynchronous communication norms and flexible schedules
• The hiring process includes technical interviews, a practical coding assessment, and conversations with the team
• Candidates should expect to coordinate across multiple time zones as part of a global and distributed team
• The team values written communication and documentation as part of their remote collaboration practices
• Justworks offers a comprehensive benefits package including health coverage for remote employees in eligible regions