Jax jobs
Explore 12 current openings. Compare roles, companies, locations, and work options, then open any job to view the full details.
Current openings
Showing 20 jobsSenior Software Engineer, JAX
NvidiaNVIDIA’s JAX team builds high-performance deep learning tools. The Senior Software Engineer designs core JAX components, drives performance, and collaborates with AI researchers to create future-proof models. Remote …
UK, RemoteRemoteFull-timeActiveMachine Learning AI Lead
NexthireSenior ML leader with 8+ years of experience building and deploying deep learning models. Holds a bachelor’s or master’s in Computer Science, AI, or Data Science and excels in Python, PyTorch, TensorFlow, and JAX. …
Bengaluru, IndiaOn-siteFull-timeActiveSenior HPC and AI Network Software Architect
NvidiaBuild scalable AI infrastructure, focusing on distributed training, real‑time inference, and communication efficiency. Develop next‑generation communication libraries (NCCL, UCX, UCC) and integrate with AI frameworks …
UK, RemoteRemoteFull-timeActiveApplication Support - Java Full Stack Developer
DXC TechnologyApplication Support - Java Full Stack Developer role at DXC Technology focuses on maintaining and enhancing multi-tier enterprise applications built with Java EE and Spring. Candidates should have 7+ years of experience …
Chennai, Tamil Nadu, IndiaOn-siteFull-timeActiveSenior Software Engineer
CubicSenior Software Engineer (Java) at Cubic Transportation Systems. Design, develop, test software for transportation solutions. Lead full SDLC: coding, unit testing, defect resolution, documentation. Mentor junior team …
Hyderabad, Telangana, IndiaHybridFull-timeActivePost-Doctoral Researcher
GoogleGoogle DeepMind India seeks a Post‑Doctoral Researcher to conduct cutting‑edge machine‑learning research, develop scalable experiments, and publish high‑impact work. The role involves collaborating with senior …
Bengaluru, Karnataka, IndiaOn-siteContractActiveSenior Software Engineer
CubicSenior Software Engineer (Fullstack) at Cubic Transportation Systems in Hyderabad. Leads design, development, and testing of enterprise applications using Java SE, Spring, Hibernate, Spring Boot, and Angular. Drives …
Hyderabad, Telangana, IndiaOn-siteFull-timeActiveSenior Engineer, AVP (JAVA)
Deutsche BankSenior Engineer, AVP (JAVA) at Deutsche Bank in Pune – Business Bay. Responsible for designing and developing end‑to‑end engineering solutions, ensuring architecture, maintainability, and testability. Leads solution …
Pune, IndiaOn-siteFull-timeActiveAssociate Gen AI Engineer Tax Technology TRS
PricewaterhouseCoopersPwC tax services focus on advising clients on tax planning, compliance, and strategy. The Generative AI Engineer will develop, implement, and optimize cutting‑edge generative models for content creation, document …
Gurugram, IndiaOn-siteFull-timeActiveJava Engineer
TATA CONSULTANCY SERVICES LTD.Technology department at TATA CONSULTANCY SERVICES LTD. builds enterprise software solutions. The Java Engineer develops RESTful APIs with JAX-RS, designs microservices using Spring Boot, and manages code with Git and …
Bengaluru, Karnataka, IndiaOn-siteFull-timeActiveJava Spring Boot Developer
CitigroupJava Spring Boot Developer with 3+ years experience in designing, developing, and maintaining microservices. Skilled in Java, Spring Boot, JAX/REST, Docker, Kubernetes, RabbitMQ, Kafka, CI/CD, and secure, scalable …
Pune, Maharashtra, IndiaHybridFull-timeActiveSoftware Engineer - Java
CubicSoftware Engineer - Java at Cubic Transportation Systems (India) is responsible for designing, developing, and testing high‑quality software solutions that support Cubic’s transportation products. The role requires 3‑5 …
Hyderabad, Telangana, IndiaOn-siteFull-timeActiveSenior Associate Gen AI Engineer Tax Technology TRS
PricewaterhouseCoopersAt PwC, the Senior Associate Gen AI Engineer in Tax Technology focuses on leveraging data analytics and generative AI to develop innovative tax solutions. The role involves designing, training, and deploying …
Gurugram, 10 C, IndiaOn-siteFull-timeActiveAI Engineer
CitigroupApplications Development team at Citigroup builds and maintains mission‑critical financial systems. The AI Engineer designs and implements Python‑based ML models, leveraging PyTorch, JAX, and Spark to enhance data …
Pune, Maharashtra, IndiaHybridFull-timeActiveResearch Scientist, AnthroKrishi, DeepMind
GoogleAnthroKrishi team at DeepMind is building a digital understanding of global agriculture to ensure a sustainable, food‑secure future. The Research Scientist will pioneer novel computer vision models on satellite data, …
Bengaluru, Karnataka, IndiaFull-timeActiveSenior Software Engineer, ML
NvidiaAI & ML Performance Team at NVIDIA builds high‑performance machine learning software that extracts maximum efficiency from NVIDIA GPUs. The Senior Software Engineer implements and optimizes ML models across domains (CV, …
Bengaluru,India,IndiaOn-siteFull-timeActiveSenior Software Engineer
CubicSenior Software Engineer (fullstack) at Cubic Transportation Systems in Hyderabad. Leads complex software assignments, mentors junior staff, and drives design, coding, testing, and deployment of enterprise applications. …
Hyderabad, Telangana, IndiaOn-siteFull-timeActiveApplied AI Frameworks Engineer
IntelData Center and AI group (DCAI) at Intel builds AI frameworks for next‑gen GPUs. The Applied AI Frameworks Engineer designs and develops HW‑agnostic and HW‑aware features for inference serving (SGLang, vLLM) and ML …
Bangalore,IndiaOn-siteFull-timeActiveAssociate Gen AI Engineer - Tax Technology TRS
PricewaterhouseCoopersPwC’s Tax Technology team seeks an Associate Gen AI Engineer in Gurugram to develop and deploy generative AI solutions for tax services. The role focuses on building state‑of‑the‑art models (LLMs, GANs, Transformers, …
Gurugram, IndiaOn-siteFull-timeActivePython Developer
Photon GroupWe are seeking a Python Developer with strong programming skills and experience in generative AI frameworks. The role requires proficiency in Python, Hugging Face Transformers, LangChain, OpenAI API, or LLamaIndex, as …
IndiaOn-siteFull-timeActive