358 Jax Jobs

As a Generative AI at Google, you will play a crucial role in designing, developing, and implementing cutting-edge AI solutions. Your deep understanding of AI/ML principles and strong engineering skills will drive the adoption and advancement of generative AI across various Google products and research initiatives. - Provide technical leadership in designing, developing, and deploying generative AI solutions. - Stay updated on the latest research and advancements in generative AI. - Identify and evaluate new generative AI technologies and tools. - Design scalable, reliable, and efficient architectures for generative AI systems. - Select and integrate appropriate generative AI models and fram...

Job Details: Job Description: We are looking for a dynamic and passionate contributors to work on Intels next generation GPUs. Works with internal engineering teams and stakeholders to deliver optimized complex math kernel operators for Intel GPU's. Partners with AI framework and workload engineers as needed to optimize endtoend AI models on Intel hardware features. Role and responsibilities for this role includes: Optimizes performance of AI models through deep knowledge and expertise of AI frameworks, algorithms, models, and related hardware. Researches, develops, and adjusts new or existing complex math kernel code to address performance, numeric issues and modify operations to enhance pe...

We are looking for a dynamic and passionate contributors to work on Intels next generation GPUs.Day-to-day work involves working on Open-source AI Frameworks such as PyTorch, Tensorflow & JAX and enabling cutting edge AI workloads (like LLMs) etc. The job role involves design and developing features for Intel' AI frameworks software stack.You will be participating in enabling and optimizing state of the art software stack for Intel's AI software stack. Role and responsibilities for this role includes: Design and develop SW features for AI frameworks - both HW-agnostic and HW-aware, especially in PyTorch device plugin and ML kernel development. Enhance and extend the Deep learning training an...

Job Description Role Purpose The purpose of the role is to create exceptional and detailed architectural application design and provide thought leadership and enable delivery teams to provide exceptional client engagement and satisfaction. Do 1. Develop architectural application for the new deals/ major change requests in existing deals a. Creates an enterprise-wide architecture that ensures systems are scalable, reliable, and manageable. b. Manages application assets and directs the development efforts within an enterprise to improve solution delivery and agility c. Guides how to construct and assemble application components and services to support solution architecture and application deve...

Job Description Min 6 to 12 years of experience in Advance JAVA, Spring Boot, Microservices and React JS development JAVA-8 or 11 Level3 : 6 years Minimum React JS with minimum 4 years Spring Boot with Microservices : 6 years Minimum Any SQL : 4 years Minimum Kafka, Docker, Kubernetes, Docker, Splunk : : 4 years Minimum SpringFrame work, JPA/Hibernate : 6 years Minimum Rest API with Spring boot:6 years Minimum GIT, Jenkins, Maven/or build tool: 6 years Minimum Mandatory Skills: JAX-RS - Java API- RESTful Web Services . Experience: 8-10 Years .

We are looking for a dynamic and passionate contributors to work on Intels next generation GPUs. Day-to-day work involves working on Open-source AI Frameworks such as PyTorch, Tensorflow & JAX and enabling cutting edge AI workloads (like LLMs) etc. The job role involves design and developing features for Intel' AI frameworks software stack. You will be participating in enabling and optimizing state of the art software stack for Intel's AI software stack. Role and responsibilities for this role includes: Design and develop SW features for AI frameworks - both HW-agnostic and HW-aware, especially in PyTorch device plugin and ML kernel development. Enhance and extend the Deep learning training ...

Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert requirements to fe...

Job Description Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert re...

Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert requirements to fe...

Job Description Role Purpose The purpose of the role is to create exceptional and detailed architectural application design and provide thought leadership and enable delivery teams to provide exceptional client engagement and satisfaction. Do 1. Develop architectural application for the new deals/ major change requests in existing deals a. Creates an enterprise-wide architecture that ensures systems are scalable, reliable, and manageable. b. Manages application assets and directs the development efforts within an enterprise to improve solution delivery and agility c. Guides how to construct and assemble application components and services to support solution architecture and application deve...

Job Description Role Purpose The purpose of the role is to support process delivery by ensuring daily performance of the Production Specialists, resolve technical escalations and develop technical capability within the Production Specialists. Do Oversee and support process by reviewing daily transactions on performance parameters Review performance dashboard and the scores for the team Support the team in improving performance parameters by providing technical support and process guidance Record, track, and document all queries received, problem-solving steps taken and total successful and unsuccessful resolutions Ensure standard processes and procedures are followed to resolve all client qu...

As a Senior Data Scientist at Auxo AI, your role will involve owning the full ML lifecycle, including model design, training, evaluation, and deployment. You will be responsible for designing production-ready ML pipelines with CI/CD, testing, monitoring, and drift detection. Additionally, you will fine-tune LLMs, implement retrieval-augmented generation (RAG) pipelines, and build agentic workflows for reasoning, planning, and decision-making. Your responsibilities will also include developing real-time and batch inference systems using Docker, Kubernetes, and Spark, as well as collaborating with product and engineering teams to integrate AI models into business applications. Furthermore, you...

We Are: At Synopsys, we drive the innovations that shape the way we live and connect Our technology is central to the Era of Pervasive Intelligence, from self-driving cars to learning machines We lead in chip design, verification, and IP integration, empowering the creation of high-performance silicon chips and software content Join us to transform the future through continuous technological innovation, You Are: You are a dynamic and forward-thinking Machine Learning Engineer with a passion for bridging advanced AI methods and the intricate world of semiconductor manufacturing You thrive in multidisciplinary environments, working at the intersection of software engineering, data science, and...

Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert requirements to fe...

Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert requirements to fe...

Role Purpose The purpose of the role is to support process delivery by ensuring daily performance of the Production Specialists, resolve technical escalations and develop technical capability within the Production Specialists. Do Oversee and support process by reviewing daily transactions on performance parameters Review performance dashboard and the scores for the team Support the team in improving performance parameters by providing technical support and process guidance Record, track, and document all queries received, problem-solving steps taken and total successful and unsuccessful resolutions Ensure standard processes and procedures are followed to resolve all client queries Resolve cl...

Role Purpose The purpose of this role is to design, test and maintain software programs for operating systems or applications which needs to be deployed at a client end and ensure its meet 100% quality assurance parameters Do 1. Instrumental in understanding the requirements and design of the product/ software Develop software solutions by studying information needs, studying systems flow, data usage and work processes Investigating problem areas followed by the software development life cycle Facilitate root cause analysis of the system issues and problem statement Identify ideas to improve system performance and impact availability Analyze client requirements and convert requirements to fe...

Role Overview: XenonStack is looking for a passionate AI Researcher with expertise in machine learning, reinforcement learning, and large language models (LLMs) to contribute to their research in Agentic AI systems. As an AI Researcher, you will collaborate with product and engineering teams to advance next-generation architectures, reasoning models, and reinforcement-based optimization for deploying and trusting AI agents in enterprise contexts. Key Responsibilities: - Conduct applied research on large language models, reasoning systems, and multi-agent architectures. - Explore and prototype reinforcement learning, retrieval-augmented generation (RAG), and hierarchical memory systems for en...

As a Senior Staff Engineer in Machine Learning Software/Firmware Engineering at Marvell, you will have a significant impact on the architecture, design, and implementation of AI infrastructure and ML compiler for Marvell's Machine Learning acceleration and infrastructure. Your responsibilities will include: - Influencing and contributing to the architecture, design, and implementation of AI infrastructure and ML compiler - Working closely with Machine learning, networking, and storage experts to develop compiler and runtime infrastructure - Creating state-of-the-art data center infrastructure solutions and IPs - Architecting, designing, and developing system software for reliable, scalable, ...

Role Overview: As a AI Model Development and Deployment Lead, your main responsibility will be to collaborate with ML Engineers and DevOps Engineers in formulating AI designs that can be built, tested, and deployed through the Route to Live and into Production using continuous integration/deployment processes. Key Responsibilities: - Model fine-tuning using open-source libraries like DeepSpeed, Hugging Face Transformers, JAX, PyTorch, and TensorFlow to improve model performance - Model deployment and maintenance, which involves deploying and managing Large Language Models (LLMs) on cloud platforms - Data management and pipeline operations, including curating and preparing training data, tran...

Role Overview: At EY, you will have the opportunity to develop a career tailored to your unique strengths and aspirations. With global support, an inclusive culture, and cutting-edge technology, you can strive to be the best version of yourself. Your distinct voice and perspective are valued to contribute to EY's continuous improvement. Join the team to create an exceptional experience for yourself and contribute to building a better working world for all. Key Responsibilities: - Develop and implement advanced AI models, encompassing generative and traditional AI, to bolster quantitative trading strategies and investment management. - Utilize machine learning algorithms to enhance BSA/AML mo...

You are seeking a highly skilled AI/ML Product Support Engineer role at GlobalLogic, transitioning from a Machine Learning development background to a client-facing, product-focused support position. Your role involves applying your technical expertise in diagnosing, resolving, and guiding the use of cutting-edge AI/ML products and services. **Key Responsibilities:** - Act as a key technical contact for customers using AI/ML products and services primarily on Cloud Computing. - Investigate, troubleshoot, and resolve product-related issues related to machine learning model performance, API integration, deployment, and infrastructure. - Guide customers on best practices for using tools like Ve...

At EY, youll have the chance to build a career as unique as you are, with the global scale, support, inclusive culture and technology to become the best version of you. And were counting on your unique voice and perspective to help EY become even better, too. Join us and build an exceptional experience for yourself, and a better working world for all. Business Consulting QAS - Quantitative Trading Book (QTB) ProfileInvestment Analyst & AI Model Development Specialist Description EY's Financial Services Office (FSO) is a dynamic, industry-focused business unit offering a broad range of integrated services that combine deep industry experience with strong functional capability and product know...

Position Requirements: 7+ years of experience in developing Java/J2EE based applications Strong experience with Spring and Spring Boot, Spring MVC, Spring Data, Spring Security Experience building REST and SOAP web services using Java Experience with Web Services (JSON, REST, JAX-RS, Spring based REST services) Experience with Java and J2EE/Java EE (including servlets, JSP, JSF, JDBC) Strong understanding of Java concurrency, concurrency patterns, experience building thread safe code Experience with messaging products like Kafka or ActiveMQ Experience with any RDBMS: Oracle, Postgres, SQL Server Experience with development tools like Eclipse, Git, Gradle, Sonar, Jenkins, Jira, Artifactory et...

Job Summary: We are looking for a talented Generative AI Developer with strong Python expertise to join our team. The ideal candidate will have hands-on experience in building and deploying AI/ML models, especially in the field of Generative AI (e.g., LLMs, Diffusion Models, Transformers). Youll work closely with data scientists, product managers, and engineering teams to integrate intelligent, generative capabilities into real-world applications. Key Responsibilities: Design, develop, and deploy Generative AI models (e.g., LLMs, GPT, diffusion models, VAEs). Fine-tune pre-trained models using Hugging Face, OpenAI, or other frameworks. Build data pipelines and APIs to support training and in...