Senior Software Engineer, JAX
Location
France + 5 moreAll locations: France | Germany | Lithuania | Poland | Switzerland | United Kingdom
Posted
42 days ago
Salary
292K - 650K / year
Seniority
Senior
Job Description
Senior Software Engineer, JAX
NVIDIA
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX. To achieve this and provide an excellent user experience, you have strong programming skills and excellent system design skills. You should have excellent communication and planning skills for this role. Come join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world! What you will be doing: - Play meaningful role in NVIDIA's effort in contributing to JAX. - Design and implement JAX core components and drive peak performance on NVIDIA products. - Work with AI applied researchers and leaders to build future-proof models - Build tools that will increase the efficiency of teams developing AI-based systems. - Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products. What we need to see: - BS in Computer Science or Computer Engineering or related field (or equivalent experience). - 6+ years relevant experience. - C/C++ and Python programming. - Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.). - Proven ability developing customer-facing solutions, balancing feature requests and bugs. - Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design. - Highly motivated with excellent verbal and written communication skills. - Ability to work successfully with multi-functional teams, principles and architects. Coordinates effectively across organizational boundaries and geographies. Ways to stand out from the crowd: - Understanding of JAX, Autograd, tracing, code generation and DSL compilers and their design. - Understanding of deep learning training in distributed contexts: multi-GPU, multi-node, synchronous vs asynchronous. - Background with software shipping cycles (dev, deploy, release, CI). - Experience building distributed systems and services at large scale. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. For Poland: The base salary range is 292,500 PLN - 507,000 PLN for Level 4, and 375,000 PLN - 650,000 PLN for Level 5.
Related Guides
Related Job Pages
More Software Engineer Jobs
C++ Developer (f/m/d)
think-cell SoftwareWe empower over 1.3 million users across 35,000 companies, including top consulting firms and major global enterprises. With offices in Berlin, Munich, Denver, Boston, London, Tokyo, and Dubai, we’re a diverse team of over 210 people, committed to innovation and excellence. think-cell is more than a job – it's a chance to be part of an innovative, inclusive team that values collaboration and growth. We encourage applicants from all backgrounds to apply. Together, we can shape the future of productivity software. We can’t wait to see what you’ll bring to the team!
At think-cell, we’re seeking bright, creative C++ Developers with a solid theoretical background. Whether you're a seasoned professional or a recent graduate in Computer Science, Physics, or Mathematics, what matters most is your problem-solving ability and C++ skills. At think-cell, we combine deep technical challenges with a healthy work-life balance and a culture that values thoughtful engineering over rushed delivery. Most of our team is based in Berlin, but we also welcome remote developers working from UTC–1 to UTC+4, reflecting our commitment to flexibility and collaboration across regions. Ideal candidates have: - Strong C++ programming skills (through education or experience) - A background in Computer Science, Mathematics, or Physics - A passion for elegant, efficient algorithms - Ability to work independently and creative problem-solving approaches - Good command of English (German not required) How You’ll Work - You’ll collaborate closely with the team, without being stuck in recurring meetings. - Your feature ships when you think it’s ready - no time pressure! - We encourage contributions from anyone What We Offer - Brilliant Team: Work with international colleagues with strong academic and professional experience - Work–life balance: Flexible working hours and an environment that respects your time and focus. - Stable Environment: Join a market leader with a reliable product. - Competitive salary - Guaranteed raise to 130k EUR after the first year Our Values As part of our team, you will embody and help shape the following principles that define us: - Forward Thinking: We embrace change and challenge the status quo. - Ownership: We take pride in our work and learn from our mistakes. - Customer First: We prioritize delivering lasting value. - Unified Team: We foster collaboration, respect, and integrity. - Strive for Excellence: We set ambitious goals and pursue quality. Why is think-cell the right place for you? We empower over 1.3 million users across 35,000 companies, including top consulting firms and major global enterprises. With offices in Berlin, Munich, Denver, Boston, London, Tokyo, and Dubai, we’re a diverse team of over 210 people, committed to innovation and excellence. think-cell is more than a job – it's a chance to be part of an innovative, inclusive team that values collaboration and growth. We encourage applicants from all backgrounds to apply. Together, we can shape the future of productivity software. We can’t wait to see what you’ll bring to the team!
• Coordinate and perform release planning, routing development, testing, and deployments on the Amazon Connect platform • Design and implement advanced call routing strategies (queues, callbacks, business hours, emergency messaging) • Develop and manage contact flows to enhance IVR and customer experience • Provide technical support and troubleshoot issues related to contact flows and scripting • Collaborate with cross-functional teams to resolve incidents and ensure smooth operations • Work closely with architecture teams to support scalable and efficient solutions
Senior Software Engineer / Team Lead - OP02108
Dev.ProSoftware Development Partner. Result-driven. Quality-obsessed.
🟢 At Dev.Pro, we work on projects that impact millions of people around the world — but we know it’s the people behind the tech who make it all happen. We truly value what makes each person unique and are building a workplace that’s inclusive, friendly, and supportive. We invite a Senior Software Engineer/Team Lead to join a strategic initiative focused on scaling a high-throughput payment orchestration platform. In this role, you’ll lead a cross-functional delivery team and work closely with client stakeholders to keep priorities clear and the team aligned. You’ll balance team leadership with hands-on coding, using your technical expertise and strong communication skills to deliver key features end-to-end while navigating ambiguity and bringing clarity to complex situations. 🟩 What's in it for you: - Work on mission-critical systems supporting high-volume financial transaction processing - Take ownership of end-to-end delivery in a highly visible, client-facing role - Provide technical leadership to a cross-functional engineering team in a fast-paced, collaborative environment, driving best practices and continuous improvement - Benefit from in-person collaboration with colleagues through occasional travel to our Zagreb office ✅ Is that you? - 5+ years of software engineering experience - Experience leading cross-functional teams of 6+ engineers, or strong tech leadership experience, with readiness to move into people leadership - Strong communication, interpersonal, and stakeholder management skills, with direct client-facing experience and ability to work effectively across diverse, cross-cultural teams - Ability to clearly explain technical concepts to both technical and non-technical audiences, align teams, and bring clarity in ambiguous situations - Demonstrated end-to-end delivery ownership, with a proactive, accountable mindset - Ability to organize work, manage priorities, and remove delivery blockers - Experience in Agile environments with exposure to modern engineering practices (automated testing, CI/CD, code reviews) - Experience delivering high-performance, reliable backend systems in mission-critical, high-throughput environments - Strong understanding of APIs, system integration, and abstraction layers - Hands-on experience with Perl, C, or Java, or a strong willingness to quickly learn Perl - Upper-Intermediate English level - Willingness to travel to our Zagreb office (twice per quarter) Desirable: - Experience in payments, fintech, card processing, or other financial services domains - Proficiency with Java-based testing frameworks - Experience with DevOps practices and building observability, logging, and resilience into products 🧩Key responsibilities and your contribution In this role, you’ll own end-to-end delivery of backend systems, lead a highly specialized cross-functional engineering team, and collaborate with client stakeholders to build and support high-performance payment solutions. - Design, develop, and maintain high-performance backend components in Perl, integrated with C-based core systems - Build mapping, customization, and business logic layers to interface with client, gateway, and host systems - Own end-to-end feature delivery, including automated testing (Java-based frameworks), code quality, and performance stability - Contribute to CI/CD pipelines, deployments, and release processes as part of delivery ownership - Build and integrate observability and monitoring features to ensure low-latency and self-recovering system operations - Participate in code reviews, architectural discussions, and knowledge-sharing sessions - Collaborate closely with cross-functional teams to deliver features from requirements to production and monitoring - Work with client stakeholders to clarify requirements, align expectations, and help resolve technical issues 🎾 What's working at Dev.Pro like? ✔️ 30 paid days off per year — take them whenever you want for vacation, holidays, or personal time ✔️ 5 paid sick days, up to 60 days of medical leave, and 6 paid days off per year for major family events like weddings, funerals, or the birth of a child ✔️ Partially covered health insurance after the probation, plus a wellness bonus for gym memberships, sports nutrition, and similar needs after 6 months ✔️ Grow your skills with English lessons, online training, conferences, and the option to explore new career paths Our next steps: ✅ Submit a CV in English — ✅ Intro call with a Recruiter — ✅ Internal interview — ✅ Client interview — ✅ Offer Interested? Find out more: 📋How we work 💻 LinkedIn Page 📈 Our website 💻IG Page
Senior Software Engineer / Team Lead - OP02108
Dev.ProSoftware Development Partner. Result-driven. Quality-obsessed.
🟢 At Dev.Pro, we work on projects that impact millions of people around the world — but we know it’s the people behind the tech who make it all happen. We truly value what makes each person unique and are building a workplace that’s inclusive, friendly, and supportive. We invite a Senior Software Engineer/Team Lead to join a strategic initiative focused on scaling a high-throughput payment orchestration platform. In this role, you’ll lead a cross-functional delivery team and work closely with client stakeholders to keep priorities clear and the team aligned. You’ll balance team leadership with hands-on coding, using your technical expertise and strong communication skills to deliver key features end-to-end while navigating ambiguity and bringing clarity to complex situations. 🟩 What's in it for you: - Work on mission-critical systems supporting high-volume financial transaction processing - Take ownership of end-to-end delivery in a highly visible, client-facing role - Provide technical leadership to a cross-functional engineering team in a fast-paced, collaborative environment, driving best practices and continuous improvement - Benefit from in-person collaboration with colleagues through occasional travel to our Zagreb office ✅ Is that you? - 5+ years of software engineering experience - Experience leading cross-functional teams of 6+ engineers, or strong tech leadership experience, with readiness to move into people leadership - Strong communication, interpersonal, and stakeholder management skills, with direct client-facing experience and ability to work effectively across diverse, cross-cultural teams - Ability to clearly explain technical concepts to both technical and non-technical audiences, align teams, and bring clarity in ambiguous situations - Demonstrated end-to-end delivery ownership, with a proactive, accountable mindset - Ability to organize work, manage priorities, and remove delivery blockers - Experience in Agile environments with exposure to modern engineering practices (automated testing, CI/CD, code reviews) - Experience delivering high-performance, reliable backend systems in mission-critical, high-throughput environments - Strong understanding of APIs, system integration, and abstraction layers - Hands-on experience with Perl, C, or Java, or a strong willingness to quickly learn Perl - Upper-Intermediate English level - Willingness to travel to our Zagreb office (twice per quarter) Desirable: - Experience in payments, fintech, card processing, or other financial services domains - Proficiency with Java-based testing frameworks - Experience with DevOps practices and building observability, logging, and resilience into products 🧩Key responsibilities and your contribution In this role, you’ll own end-to-end delivery of backend systems, lead a highly specialized cross-functional engineering team, and collaborate with client stakeholders to build and support high-performance payment solutions. - Design, develop, and maintain high-performance backend components in Perl, integrated with C-based core systems - Build mapping, customization, and business logic layers to interface with client, gateway, and host systems - Own end-to-end feature delivery, including automated testing (Java-based frameworks), code quality, and performance stability - Contribute to CI/CD pipelines, deployments, and release processes as part of delivery ownership - Build and integrate observability and monitoring features to ensure low-latency and self-recovering system operations - Participate in code reviews, architectural discussions, and knowledge-sharing sessions - Collaborate closely with cross-functional teams to deliver features from requirements to production and monitoring - Work with client stakeholders to clarify requirements, align expectations, and help resolve technical issues 🎾 What's working at Dev.Pro like? ✔️ 30 paid days off per year — take them whenever you want for vacation, holidays, or personal time ✔️ 5 paid sick days, up to 60 days of medical leave, and 6 paid days off per year for major family events like weddings, funerals, or the birth of a child ✔️ Partially covered health insurance after the probation, plus a wellness bonus for gym memberships, sports nutrition, and similar needs after 6 months ✔️ Grow your skills with English lessons, online training, conferences, and the option to explore new career paths Our next steps: ✅ Submit a CV in English — ✅ Intro call with a Recruiter — ✅ Internal interview — ✅ Client interview — ✅ Offer Interested? Find out more: 📋How we work 💻 LinkedIn Page 📈 Our website 💻IG Page

