Find a job 
Featured Companies 
Featured Universities 
For Companies At a Glance
- Tasks: Develop and optimise AI frameworks, focusing on PyTorch integrations for a custom AI accelerator.
- Company: Join an innovative semiconductor team at the forefront of GPU and AI technology.
- Benefits: Enjoy hybrid work options and the chance to work with cutting-edge technology.
- Why this job: Be part of a team shaping the future of AI with impactful projects and collaboration.
- Qualifications: 5+ years in AI/ML software development; strong skills in C++, CUDA, or OpenCL required.
- Other info: Opportunity to work closely with hardware teams and contribute to high-performance AI solutions.
The predicted salary is between 43200 - 72000 £ per year.
An innovative semiconductor team is seeking a Software Engineer with deep expertise in AI and machine learning frameworks to help build the next generation of GPU and AI acceleration solutions. This is a unique opportunity to contribute to cutting-edge technology focused on delivering high-performance and energy-efficient compute platforms for modern AI workloads.
You will be working on a flagship GPU and AI platform supporting PyTorch, OpenCL, and Vulkan, designed to bring scalable, efficient AI capabilities to developers and researchers across the industry.
Role Overview:
As a Software Engineer – AI Framework, you will be responsible for developing and optimizing PyTorch integrations and other AI tools for a custom AI accelerator platform. You will work closely with hardware and software teams to ensure tight integration, performance tuning, and a seamless developer experience.
Key Responsibilities:
- Develop and maintain PyTorch integration for a custom AI platform.
- Build and optimize kernels and libraries for core AI operations.
- Create tooling for neural network deployment and optimization.
- Benchmark, analyze, and improve AI workload performance.
- Collaborate with the hardware team to guide architectural decisions.
- Extend support to additional frameworks (e.g., TensorFlow, ONNX).
- Produce developer documentation and resources.
Requirements:
- 5+ years of experience in AI/ML software development.
- Deep understanding of PyTorch internals and other major ML frameworks.
- Experience optimizing deep learning performance on accelerator hardware.
- Solid knowledge of deep learning algorithms and compute patterns.
- Strong programming skills in C++, CUDA, or OpenCL.
- Background in performance profiling and optimization.
- BS/MS in Computer Science, Electrical Engineering, or a related field.
Interested? Send your CV to apply.
Software Engineer - AI Framework employer: microTECH Global Limited
Contact Detail:
microTECH Global Limited Recruiting Team
StudySmarter Expert Advice 🤫
We think this is how you could land Software Engineer - AI Framework
✨Tip Number 1
Familiarise yourself with the latest advancements in AI and machine learning frameworks, especially PyTorch. Being able to discuss recent updates or features during your interview can demonstrate your passion and commitment to the field.
✨Tip Number 2
Engage with the developer community by contributing to open-source projects related to AI frameworks. This not only enhances your skills but also showcases your ability to collaborate and innovate, which is crucial for the role.
✨Tip Number 3
Prepare to discuss specific projects where you've optimised deep learning performance on accelerator hardware. Real-world examples will help you stand out and show how your experience aligns with our needs.
✨Tip Number 4
Network with professionals in the AI and semiconductor industries. Attend relevant conferences or webinars to make connections and gain insights that could be beneficial during your application process.
We think you need these skills to ace Software Engineer - AI Framework
Some tips for your application 🫡
Tailor Your CV: Make sure your CV highlights your experience in AI and machine learning frameworks, particularly with PyTorch and other relevant technologies. Emphasise your programming skills in C++, CUDA, or OpenCL, and any specific projects that showcase your expertise.
Craft a Compelling Cover Letter: Write a cover letter that explains why you are passionate about AI and how your background aligns with the role. Mention specific experiences where you've developed or optimised AI tools, and express your enthusiasm for contributing to cutting-edge technology.
Showcase Relevant Projects: Include a section in your application that details relevant projects or contributions to open-source initiatives related to AI frameworks. This can demonstrate your hands-on experience and problem-solving skills in real-world scenarios.
Highlight Collaboration Skills: Since the role involves working closely with hardware and software teams, make sure to mention any past experiences where you collaborated with cross-functional teams. Highlight your ability to communicate technical concepts effectively to non-technical stakeholders.
How to prepare for a job interview at microTECH Global Limited
✨Showcase Your Technical Skills
Be prepared to discuss your experience with AI and machine learning frameworks, especially PyTorch. Highlight specific projects where you've optimised performance or developed integrations, as this will demonstrate your hands-on expertise.
✨Understand the Company’s Technology
Research the company's GPU and AI acceleration solutions. Familiarise yourself with their products and how they integrate with frameworks like OpenCL and Vulkan. This knowledge will help you ask insightful questions and show your genuine interest.
✨Prepare for Problem-Solving Questions
Expect technical questions that assess your problem-solving skills in AI/ML contexts. Practice coding challenges related to deep learning algorithms and performance optimisation, as these are likely to come up during the interview.
✨Demonstrate Collaboration Skills
Since the role involves working closely with hardware and software teams, be ready to discuss your experience in collaborative environments. Share examples of how you've successfully worked with cross-functional teams to achieve project goals.
