Software engineer python tensorflow Work, Jobs (with Salaries)

Staff Engineer (ML Engineer)
Graphcore —London
- Machine learning
- Linux
- Python
- Annual leave
4d
AI Solutions Engineer - Vice President - London
Goldman Sachs —London
- Full-time
- Azure
- Computer Science
- Engineering
- Employee assistance programme
Machine Learning Engineer
BeyondMath —London
- Full-time
- Hybrid work
- Doctoral degree
- Computer Science
- Master's degree
Quick apply
Key Customers Solutions Architect EMEA
Nebius —London
- Remote
- Azure
- AWS
- Project management
Quick apply
Lead Software Engineer
WeBuild-AI —London
- Hybrid work
- Azure
- System design
- Java
Quick apply
Senior Machine Learning Engineer
AECOM —London
- Full-time
- Doctoral degree
- Computer Science
- Engineering
- Employee stock purchase plan
- Annual leave
AI Engineering Lead
Shawbrook —London
- Full-time
- Azure
- AWS
- Machine learning
- Employee assistance programme
- Employee discount
AI Engineer
Version 1 —Newcastle upon Tyne
- Full-time
- Hybrid work
- Azure
- DevOps
- Master's degree
- Employee assistance programme
- Employee discount
Quick apply
6d
Principal Solutions Architect
Nebius —London
- Remote
- Azure
- AWS
- AI
Quick apply
Foundation Model Engineer
CommonAI C.I.C. —Cambridge
- Full-time
- Hybrid work
- Master's degree
- Maths
- Machine learning
Quick apply
Senior Applied Scientist, Insights, Prime Video
Amazon Development Centre (London) Limited —London
- Full-time
- Doctoral degree
- Engineering
- Java
External - AI Architect (m/f/*)
Solventum —Loughborough
- Full-time
- Computer Science
- Master's degree
- SQL
Business Consultant, AI & NLP
LSEG (London Stock Exchange Group) —London
- Full-time
- Sales
- AWS
- Bachelor's degree
Senior Principal AI Engineer
Expleo —London
- Azure
- AWS
- AI
Machine Learning Engineer
Trustpilot —London
- Azure
- AWS
- Machine learning
- Employee assistance programme
- Employee discount
Associate Director, Lead AI Architect, Security & Justice, Engineering, AI & Data, Technology & Transformation
Deloitte —Edinburgh
- Hybrid work
- Azure
- Doctoral degree
- Computer Science
Principal Data & AI Trainer
QA —London
- Azure
- Doctoral degree
- Computer Science
- Annual leave
- Employee assistance programme
Forward-Deployed Data Scientist II
Braze —London
- Doctoral degree
- Computer Science
- DevOps
- Employee stock purchase plan
- Annual leave
Quick apply
12d
Senior Research Engineer
IXICO —London
- Full-time
- Hybrid work
- Doctoral degree
- Presentation skills
- Machine learning
- Annual leave
- Employee assistance programme
11d
Senior Dev/ML Ops Engineer
Humanoid —London
- Full-time
- Azure
- DevOps
- AWS
- Annual leave
- Sick pay
Quick apply

I want to receive the latest job alerts for software engineer python tensorflow

By signing in to your account, you agree to SimplyHired's Terms of Service and consent to our Cookie and Privacy Policy.

Staff Engineer (ML Engineer)

Graphcore -
London

Apply Now

Job details

4 days ago

Benefits

Private dental insurance
Annual leave
Discounted or free food
Company pension
Private medical insurance

Qualifications

TensorFlow
PyTorch
Machine learning
Linux
Python
Debugging

Full job description

About the job

Validate the ML stack that turns accelerator hardware into trusted AI performance.

This role sits where modern ML models meet Graphcore's software and hardware stack. You will test, benchmark and validate complex systems before they reach customers.

Your work will expose regressions, correctness issues and performance limits across frameworks, models and execution environments. You will help teams understand what is working, what is breaking and why.

You will run open source models, build automated benchmarking pipelines and create targeted tests for low level ML behaviour. That includes numerical precision, quantisation, attention mechanisms, distributed execution and model subgraphs.

This is a role for someone who wants to stay close to how AI systems really work. You will not design new models, but you will make them run reliably on ambitious infrastructure.

The team and culture

The ML QA team is where Graphcore's ML software stack comes together for validation. Work spans unit tests, full model benchmarks, distributed workloads, simulators, emulators and hardware targets.

Engineers are expected to take ownership, question assumptions and improve how testing is done. The team moves quickly, but decisions are grounded in evidence, benchmark data and technical discussion.

You will work closely with software, infrastructure and hardware teams. Squads organise around priorities, with space for engineers to shape roadmaps and raise the quality bar.

What we're looking for

Strong experience in Machine Learning or ML-adjacent software engineering roles.
A solid grasp of neural networks, training, inference, numerical precision and performance trade-offs.
Hands-on experience with PyTorch, TensorFlow, JAX, Triton or similar ML frameworks and tools.
Strong Python skills for automation, experimentation, benchmarking and reporting.
Experience designing, running and analysing ML benchmarks or model experiments.
Confident debugging skills in Linux, with curiosity about model behaviour and system performance.

Benefits

Unlimited annual leave
Up to 5% matched pension
Phantom equity – share in Graphcore's success
True flexibility in how and where you work
Office spaces designed for collaboration
Free food and an on-site barista
Health cash plan
Income protection
Life assurance
Along with other benefits you can choose from (private medical insurance, dental plan etc)

We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.

We're an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we're happy to support you.

Join the Team at Graphcore

Graphcore is one of the world's leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry.

As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world's most transformative technologies. Together, they share a bold vision: to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone.

Graphcore's teams are drawn from diverse backgrounds and bring a broad range of skills and perspectives. A melting pot of AI research specialists, silicon designers, software engineers and systems architects, Graphcore brings together deep expertise to solve complex problems and deliver meaningful progress in AI compute.

If you want to validate the systems behind next generation AI compute, we'd love to hear from you. Apply now to help build what comes next.

Apply Now

Refine Your Search

software engineer python tensorflow jobs

Staff Engineer (ML Engineer)

AI Solutions Engineer - Vice President - London

Machine Learning Engineer

Key Customers Solutions Architect EMEA

Lead Software Engineer

Senior Machine Learning Engineer

AI Engineering Lead

AI Engineer

Principal Solutions Architect

Foundation Model Engineer

Senior Applied Scientist, Insights, Prime Video

External - AI Architect (m/f/*)

Business Consultant, AI & NLP

Senior Principal AI Engineer

Machine Learning Engineer

Associate Director, Lead AI Architect, Security & Justice, Engineering, AI & Data, Technology & Transformation

Principal Data & AI Trainer

Forward-Deployed Data Scientist II

Senior Research Engineer

Senior Dev/ML Ops Engineer

I want to receive the latest job alerts for software engineer python tensorflow

Related Searches

Explore jobs in more locations

About the job

The team and culture

What we're looking for

Benefits

Join the Team at Graphcore

Jobseeker tools

Employer Tools

Browse

Stay Connected