Senior Software Engineer - AI Compute Libraries & Performance

Graphcore

Salary Range: PLN 260,400 - 352,200 Subject to alignment to the responsibilities and duties of the role. ABOUT GRAPHCORE  At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to infrastructure at datacenter scale. As part of the SoftBank Group, backed by significant long-term investment, we are delivering key technology into the fast-growing SoftBank AI ecosystem. To meet the vast and exciting AI opportunity, Graphcore is expanding its teams around the world. We are bringing together the brightest minds to solve the toughest problems, in a place where everyone has the opportunity to make an impact on the company, our products and the future of artificial intelligence. Job Summary   As a Senior Software Engineer, you’ll build high-performance AI compute libraries for next-generation AI hardware, working on linear algebra, tensor operations, numerical correctness, and low-level performance optimisation. The Team  This is an exciting opportunity to join an expanding team at Graphcore. Our ML Kernels & Runtime team are responsible for delivering high performance compute library to help customers gain the maximum performance from AI hardware.  Responsibilities and Duties  * Design and implement kernels for linear algebra and tensor ops (GEMM, batched GEMM, convolutions, reductions, elementwise and fused operations) in C++  * Own performance and correctness - add microbenchmarks, regression tests, numerics validation  * Profile and optimise across for next generation of AI hardware - threading, cache locality, memory layout, and kernel launch efficiency.  * Debug issues, resolve bugs and generally improve the quality and functionality of the product  * Actively engage in and support Agile ways of working within the team  * Mentor colleagues within the team, sharing knowledge and providing guidance where appropriate  Candidate Profile     Essential  * Excellent programming and scripting skills using C++ and Python  * Understanding of processor architectures and profiling on Linux  * Possess excellent written and oral communication skills, good work ethics, high sense of team-work  * Love to produce quality work and be a team player  Desirable  * Strong command of algorithmic performance - vectorisation, memory hierarchy, threading, lock-free patterns  * Hands-on with at least one BLAS/DNN stack and able to read/extend kernels  * Comfort with CPU micro-optimisations and numerical stability/trade-offs across FP32/FP16/BF16/FP8  * Experience integrating native code into PyTorch or similar (custom ops, extensions, dispatch keys)  * ABI/API stability and packaging for Linux system (manylinux, wheels)  BENEFITS In addition to a competitive salary, Graphcore offers annual leave policy, medical and dental health plans, a gym card, and employee pension (matched up to 4%). We review our benefits on a yearly basis to ensure we offer a valuable and rewarding benefits programme to our employees. We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We offer an equal opportunity process and understand that there are visible and invisible differences in all of us. We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments.

Databricks

P-78 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best data and AI infrastructure platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to solve technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As a software engineer with a backend focus, you will work with your team to build infrastructure and products for the Databricks platform at scale. The impact you'll have: Our backend teams span many domains, from our core compute fabric resource management infrastructure, to service platforms, to machine learning infrastructure. For instance, you might work on challenges such as: * Resource management infrastructure powering the big data and machine learning workloads on the Databricks platform in a scalable, secure, and cloud-agnostic way * Develop reliable, scalable services and client libraries that work with massive amounts of data on the cloud, across geographic regions and Cloud providers * Build tools to allow Databricks engineers to operate their services across different clouds and environments * Build services, products and infrastructure at the intersection of machine learning and distributed systems. What we look for: * BS (or higher) in Computer Science, or a related field * 5+ years of production level experience in one of: Java, Scala, C++, or similar language. * Experience developing large-scale distributed systems. * Experience working on a SaaS platform or with Service-Oriented Architectures. * Experience with cloud technologies, e.g. AWS, Azure, GCP, Docker, or Kubernetes. * Experience with security and systems that handle sensitive data. * Good knowledge of SQL. Benefits * Comprehensive health coverage including medical, dental, and vision * 401(k) Plan * Equity awards * Flexible time off * Paid parental leave * Family Planning * Gym reimbursement * Annual personal development fund * Work headphones reimbursement * Employee Assistance Program (EAP) * Business travel accident insurance   Pay Range Transparency Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles.  Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here.   Local Pay Range $166,000—$225,000 USD About Databricks Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter, LinkedIn and Facebook. Benefits At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here. Our Commitment to Diversity and Inclusion At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics. Compliance If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

PhysicsX

ABOUT US PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across the entire engineering lifecycle, PhysicsX unlocks new levels of optimization and automation in design, manufacturing, and operations — empowering engineers to push the boundaries of possibility. Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive. NOTE: WE ARE CURRENTLY RECRUITING FOR MULTIPLE POSITIONS ACROSS DIFFERENT LEVELS, HOWEVER PLEASE ONLY APPLY FOR THE ROLE THAT BEST ALIGNS WITH YOUR SKILLSET AND CAREER GOALS. WHAT YOU WILL DO  * Shape Research group strategy and culture in a significant way, especially in domains of expertise. * Be opinionated and formulate strategy on engineering topics relevant to our Research priorities, especially on: scaled engineering, securing compute, infrastructure stack. * Define necessary profiles to execute this strategy. * Promote effective working patterns and proactively flag issues with team dynamics to foster a productive environment. * Nurture younger colleagues to grow their skillset and guide their professional development. * Own Research work-streams at a high-level to deliver outcomes. * Align priorities with problem stakeholders, internal and external. * Set the technical direction for the stream and apply judgement and taste to drive progress. * Plan roadmaps with clear milestones for key decisions and outcomes. * Organise and guide the more junior members of the team to effectively execute and deliver against this roadmap. * Communicate purpose and key outcomes to raise awareness across the company and create opportunities for use and deployment. * The below activities in particular. * Work closely with our research scientists and simulation engineers to build and deliver models that address real-world physics and engineering problems. * Design, build and optimise machine learning models with a focus on scalability and efficiency in our application domain. * Transform prototype model implementations to robust and optimised implementations. * Implement distributed training architectures (e.g., data parallelism, parameter server, etc.) for multi-node/multi-GPU training and explore federated learning capacity using cloud (e.g., AWS, Azure, GCP) and on-premise services. * Work with research scientists to design, build and scale foundation models for science and engineering; helping to scale and optimise model training to large data and multi-GPU cloud compute. * Identify the best libraries, frameworks and tools for our modelling efforts to set us up for success. * Discuss the results and implications of your work with colleagues and customers, especially how these results can address real-world problems. * Work at the intersection of data science and software engineering to translate the results of our Research into re-usable libraries, tooling and products. * Foster a nurturing environment for colleagues with less experience in ML / Engineering for them to grow and you to mentor.   WHAT YOU BRING TO THE TABLE * Enthusiasm about developing machine learning solutions, especially deep learning and/or probabilistic methods, and associated supporting software solutions for science and engineering. * Ability to work autonomously and scope and effectively deliver projects across a variety of domains. * Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly. * Excellent collaboration and communication skills — with teams and customers alike. * MSc or PhD in computer science, machine learning, applied statistics, mathematics, physics, engineering, software engineering, or a related field, with a record of experience in any of the following: * scientific computing; * high-performance computing (CPU / GPU clusters); * parallelised / distributed training for large / foundation models. *  4 years of experience in a data-driven role in a professional industry setting, where you have been instrumental in most of the below: * * scaling and optimising ML models, training and serving foundation models at scale (federated learning a bonus); * employing distributed computing frameworks (e.g., Spark, Dask) and high-performance computing frameworks (MPI, OpenMP, CUDA, Triton); * employing cloud computing (on hyper-scaler platforms, e.g., AWS, Azure, GCP); * building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; * building or using C/C++ for computer vision, geometry processing, or scientific computing; * following and promoting software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps); * container-izing and orchestrating compute tasks (Docker, Kubernetes, Slurm); * writing pipelines and experiment environments, including running experiments in pipelines in a systematic way. WHAT WE OFFER Build what actually matters Help shape an AI-native engineering company at a formative stage, tackling problems that genuinely matter for industry and society. This is work with real-world impact - and something you can be proud to stand behind. Learn alongside exceptional people Work with a high-caliber, collaborative team of engineers, scientists, and operators who care deeply about doing great work, and about helping each other get better. We come from diverse backgrounds, but we share a commitment to operating at the highest level and addressing some of the most complex challenges out there. If you’re ambitious, thoughtful, and driven by impact, you’ll feel at home. Influence over hierarchy We operate with a flat structure: good ideas win - wherever they come from. Questioning assumptions and challenging the status quo isn’t just welcomed, it’s expected. Sustainable pace, long-term ambition Building meaningful technology is a marathon, not a sprint. We believe in balancing focused, ambitious work with a life beyond it. Our hybrid model blends time together in our Shoreditch office with work-from-home days, giving you the flexibility to work sustainably while staying connected in person. And it doesn’t stop there … 🚀 Equity options - share meaningfully in the company you’re helping to build. 🏦 10% employer pension contribution - because investing in future matters. 🍽️ Free office lunches - to keep you energised and focused. 👶 Enhanced parental leave - 3 months full pay paternity and 6 months full pay maternity leave, to provide extra flexibility during the moments that matter most. 🍼 YellowNest nursery scheme - to help working parents manage childcare costs. ☀️ 25 days of Annual Leave (+ Public Holidays) - because taking time to rest matters. 🏥 Private medical insurance - 100% employee cover, giving you complete peace of mind. 💪 Wellhub Subscription - gain access to thousands of gyms, classes and wellness apps, supporting both physical and mental wellbeing. 👀 Eye tests - because good work depends on good health. 📈 Personal development - dedicated support for learning, development, and leveling up over time. 💛 Employee Assistance Programme (EAP) - confidential wellbeing support, available whenever you need it. 🚲 Bike2Work scheme and 🚆 Season ticket loan - to make getting to work easier and greener. 🚗 Octopus EV salary sacrifice - for a simpler, more sustainable way to drive electric. 🔎 Watch this space, we’re continuing to build this as we grow…   We value diversity and are committed to equal employment opportunity regardless of sex, race, religion, ethnicity, nationality, disability, age, sexual orientation or gender identity. We strongly encourage individuals from groups traditionally underrepresented in tech to apply. To help make a change, we sponsor bright women from disadvantaged backgrounds through their university degrees in science and mathematics.    We collect diversity and inclusion data solely for the purpose of monitoring the effectiveness of our equal opportunities policies and ensuring compliance with UK employment and equality legislation. This information is confidential, used only in aggregate form, and will not influence the outcome of your application.