Senior Software Engineer (Prediction Ml)

Who are we?

Based in Oxford and with offices in Canada and the US, Oxa is an international startup founded in 2014 and scaling up rapidly (315+ Oxbots and growing).

As the global leader in our industry, we’re fuelled by a bold purpose: to make the Earth move better. From passenger shuttles to industrial vehicles, our operating system for Universal Autonomy™ is transforming the way people and goods are transported by enabling any vehicle, in any environment, to operate autonomously — safely, securely and efficiently.

Our technology is capturing investors’ imagination. To date we’ve raised $140 million in our Series C investment round. Accommodating growing demand from new and current shareholders, this funding is driving our expansion in North America and EMEA, and accelerating the deployment of our technology in domains where there is both urgent need and potential to scale.

Your Team

You will join a growing group of Prediction experts reporting to the Prediction Lead, and work with a multidisciplinary team of world-class experts in computer science, AI, robotics and machine learning to drive our Universal Autonomy™ platform forward. Your team will leverage machine learning and data to train and deploy powerful on-vehicle prediction models, enabling Oxa Driver™ to plan sophisticated and interactive driving behaviours scalably across all of our customer domains.

Your Role

As our Senior Software Engineer (Prediction - ML), you will be responsible for collaborating with your team to develop and integrate machine learning models for interactive intent and motion forecasting, while remaining aligned to business needs and long-term architectural strategy. You will also be responsible for engaging with team members and the broader engineering community within Oxa to share knowledge and insights, as well as providing meaningful feedback and mentorship to team members, to create an environment that supports collaboration and mutual understanding.

You will:

• Contribute significantly to the engineering lifecycle, including scoping, design, implementation, testing, deployment, and maintenance of your team's work.
• Contribute to the development of high quality APIs that allow prediction models to be used as a building block for on-vehicle capabilities.
• Document and train others on your team's work, focus on improving features, and understand the business and customer value of your team's work.
• Assist in the development of suitable heuristics and guardrails for learned prediction models.
• Influence technical strategy for Prediction, ensuring alignment with the Prediction Lead and your counterparts in other teams.
• Contribute to hiring and onboarding at scale by defining role responsibilities and requirements, leading interviews, engaging in recruiting outreach for your team, and mentoring and developing new team members.
• Positively impact product quality, customer experience, and maintainability, and take on projects that improve team processes and solutions.
• Work with customers, stakeholders, and peers to ensure the right solutions are delivered, and actively contribute to resolving disagreements through constructive dialog.
• Create appropriate metrics and tools to measure and benchmark the performance of learned prediction models.
• Collaborate with your team to develop, integrate and deploy state of the art machine learning algorithms for interactive motion prediction on autonomous vehicles.

Requirements

What you need to succeed:

• Deep learning and model architectures
• Probability and graphical models
• Prior experience working with state-of-the-art time series prediction models, graph neural networks and/or transformers.
• Optimization and loss function design
• Ability to address problems with visible risks and roadblocks while bringing clarity to problems and identifying simple solutions.
• A proven track record of making effective priority trade-offs between new feature development and operational work.
• A willingness to embrace change and ambiguity, reflecting on challenges and applying learnings to improve and grow.
• Fluency in Python and working knowledge of modern C++.
• Familiarity with Git or other version control systems.
• An appreciation for the role of safety, security and ethics in robotics and AI.
• Hyperparameter selection
• Training and validation methodology, including ablation studies
• Common software frameworks (e.g., PyTorch, TensorFlow, etc.)
• Strong technical communication skills including code review and documentation.
• Model deployment (e.g., ONNX)
• A deep understanding of Machine Learning fundamentals including:
• A willingness to own your team’s deliverables and independently drive them forward.
• Demonstrated operational excellence in all of your work.
• Data pre-processing
• An understanding of the business and customer value of the technology you work on - we invent on behalf of our customers.

Extra kudos:

• Exposure to Perception, Planning and Control algorithms.
• PhD/MSc in Computer Science or a related field (e.g., Robotics, Engineering, Physics).
• Previous years of relevant professional experience.
• Prior experience developing and fielding autonomous systems.
• Relevant publications and/or talks at industry conferences or in journals.
• Familiarity with open benchmark datasets (e.g., Waymo Open Motion Dataset, nuScenes/nuPlan, Argoverse, Lyft Level 5, etc.)

The Candidate Journey: Multi-Step and Two-Way

No-one wants to feel like a square peg in a round hole, so this process is designed to give you every chance to get the measure of us, and us of you. The various stages give you every opportunity to show your unique strengths and qualities, and enables each of us to establish if we’re a good fit for the other. If the fit is good and you’re selected, you’re then in a position to do great work and thrive, which is what everyone wants.

Benefits

We provide:

• Extended Health Care
• Critical Illness Cover
• AD&D Insurance
• Dental Cover
• Personal Spending Account
• Employee Assistance Program
• Life Insurance
• Long Term Disability Insurance
• Competitive salary, benchmarked against the market and reviewed annually
• Hybrid working arrangements
• Health Spending Account

Our Culture:

We promote an open and inclusive culture that empowers our Oxbots to bring their whole, authentic selves to work every day. Oxa is proud to be an inclusive organisation and, as such, we require all team members within our recruitment process to understand and deploy best practices focused on de-biasing the whole recruitment cycle. We also apply a neuro inclusive lens to our recruitment process and want each potential Oxbot to enjoy the best experience possible for them. Please share with us any individual needs or reasonable adjustments we may need to make in advance of commencing the interview process with us.

Learn more about our culture here

Why become an Oxbot?

Our team of engineers, mathematicians and experts in AI, machine learning and much else is truly world-class. They are solving the most exciting and important technological challenges of our times.

But as well as smarts, Oxbots have heart. Our diverse, multi-cultural crew is guided by a shared vision to bring the myriad benefits of autonomy to our customers and partners. And in a company that celebrates uniqueness as much as skill and experience, we do it with energy, conviction and a healthy dose of excitement too. If you are bold, creative and hyper skilled, come and create the future of autonomy with us at Oxa.