Senior Software Engineer, AI (Remote UK)
WFA Digital Insight
As the demand for AI and machine learning specialists continues to grow, with a 25% increase in 2025, companies like Justworks are at the forefront of innovation. With a strong focus on digital skills and remote work, this Senior Software Engineer, AI role offers a unique opportunity to build AI agents that transform operations. Candidates with a passion for AI and software engineering will find this role intriguing, especially given Justworks' supportive and entrepreneurial environment. Before applying, candidates should be prepared to showcase their ability to work autonomously and collaboratively in a fast-paced environment.
Job Description
About the Role
Justworks is seeking a highly skilled Senior Software Engineer, AI to join their team in building AI agents that fundamentally transform how operations work at scale. This role entails working on a small, highly autonomous cross-functional team to develop an agent platform that can read operational knowledge, reason about it in context, and execute tasks end-to-end. The ideal candidate will have a strong passion for building AI-native systems and be able to take ownership of large segments of the technology stack.The role matters because it has the potential to significantly impact the customer experience by providing scalable and efficient solutions. The team context is collaborative, with a focus on innovation and experimentation. The reporting structure is flat, allowing for direct communication and feedback.
What You Will Do
- Design, build, and own core components of the agent platform, from the orchestration layer to the tool integrations connecting it to internal systems.
- Build and evolve the capabilities layer: APIs, data access patterns, and service integrations for agents to execute operational workflows.
- Architect the knowledge and memory infrastructure, allowing agents to retrieve the data and act across our systems.
- Design and implement the trust and safety framework, including audit trails, human-in-the-loop controls, and guardrails for safe and auditable AI-driven operations.
- Contribute to agent evaluation and testing infrastructure, defining how we measure accuracy, reliability, and regression across agent behaviors.
- Partner directly with Operations to understand workflows, identify automation opportunities, and translate process knowledge into agent capabilities.
- Mentor other engineers on the team as it grows.
- Collaborate with cross-functional teams to ensure seamless integration of AI agents with existing systems.
- Participate in code reviews and contribute to the improvement of the overall codebase.
What We Are Looking For
- 5+ years of experience in software engineering, with a focus on AI and machine learning.
- Strong proficiency in programming languages such as Python, Java, or C++.
- Experience with AI and machine learning frameworks such as TensorFlow or PyTorch.
- Knowledge of cloud-based platforms such as AWS or Google Cloud.
- Strong understanding of data structures and algorithms.
- Excellent problem-solving skills and ability to work autonomously.
- Strong communication and collaboration skills.
- Experience with agile development methodologies.
Nice to Have
- Experience with containerization using Docker.
- Knowledge of Kubernetes or other orchestration tools.
- Familiarity with CI/CD pipelines.
- Experience with testing frameworks such as Jest or Pytest.
Benefits and Perks
- Competitive salary and benefits package.
- Opportunity to work on cutting-edge AI and machine learning projects.
- Collaborative and dynamic work environment.
- Flexible working hours and remote work options.
- Professional development opportunities.
- Access to the latest tools and technologies.
- Recognition and rewards for outstanding performance.
How to Stand Out
- When applying, make sure to highlight your experience with AI and machine learning frameworks, as well as your ability to work autonomously and collaboratively.
- Be prepared to showcase your problem-solving skills and ability to think critically.
- Familiarize yourself with Justworks' company values and mission to demonstrate your alignment with the company's culture.
- Make sure your resume and cover letter are tailored to the role and highlight your relevant skills and experience.
- Prepare to discuss your experience with agile development methodologies and your ability to adapt to changing priorities.
- Research the company and the role to demonstrate your interest and enthusiasm for the position.
This is a remote position listed on WFA Digital, the platform for professionals who work from anywhere. Browse more remote jobs across all categories.