Senior Software Engineer (Prediction ML)

Key Skills

C++, Deep Learning, Design, Effective communication skills, Git, Machine learning techniques, ONNX, Optimization, Python Programming, PyTorch, Software Development, TensorFlow

Job Description

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:

• Collaborate with your team to develop, integrate and deploy state of the art machine learning algorithms for interactive motion prediction on autonomous vehicles.
• Assist in the development of suitable heuristics and guardrails for learned prediction models.
• Contribute to the development of high quality APIs that allow prediction models to be used as a building block for on-vehicle capabilities.
• Create appropriate metrics and tools to measure and benchmark the performance of learned prediction models.
• Work with customers, stakeholders, and peers to ensure the right solutions are delivered, and actively contribute to resolving disagreements through constructive dialog.
• Contribute significantly to the engineering lifecycle, including scoping, design, implementation, testing, deployment, and maintenance of your team's work.
• Influence technical strategy for Prediction, ensuring alignment with the Prediction Lead and your counterparts in other teams.
• 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.
• Positively impact product quality, customer experience, and maintainability, and take on projects that improve team processes and solutions.
• 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.

Requirements

What you need to succeed:

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

Extra kudos:

• PhD/MSc in Computer Science or a related field (e.g., Robotics, Engineering, Physics).
• Previous years of relevant professional experience.
• 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.)
• Prior experience developing and fielding autonomous systems.
• Exposure to Perception, Planning and Control algorithms.

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:

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