Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Software Engineer, AI/ML & GPU Serving

$61k - $101k
Full-time

J.P. Morgan

Salary: $61,000 - 101,000 per year Requirements:

  • Formal training or certification in software engineering concepts, with at least 3 years of applied experience
  • Professional software development experience, particularly with machine learning systems
  • Hands-on experience with enterprise-authorized AI-assisted development tools, including critically evaluating and validating generated outputs for correctness, performance, and security
  • Ability to use AI responsibly in engineering workflows, protect sensitive data, handle inputs and outputs securely, follow resiliency and security expectations, and guide peers in safe practices
  • Experience designing and developing scalable systems and services, such as web services or APIs
  • Proficiency in Python and experience with machine learning frameworks such as TensorFlow or PyTorch
  • Experience with cloud-native technologies such as Docker and Kubernetes
  • Experience with a public cloud platform such as AWS or GCP
  • Knowledge of system design fundamentals, including reliability, scalability, and performance trade-offs
  • Experience with distributed systems and microservices architectures
  • Experience with Cassandra or an equivalent NoSQL database
  • Preferred: Working knowledge of additional languages such as Java or C++
  • Preferred: Exposure to deploying or operating GPU workloads in Kubernetes environments
  • Preferred: Exposure to machine learning model-serving frameworks such as TorchServe, TensorFlow Serving, or Triton Inference Server
  • Preferred: Familiarity with distributed training and inference systems
  • Preferred: Awareness of model compression techniques and edge deployment concepts
  • Preferred: Familiarity with API management, Domain-Driven Design, and Agile development environments
Responsibilities:
  • Design, develop, and troubleshoot software solutions, applying creative approaches to resolve complex technical problems
  • Build secure, high-quality production code and maintain algorithms that operate in coordination with relevant systems
  • Use authorized AI coding tools to improve code quality, delivery speed, and productivity; validate their outputs through peer review, automated testing, and secure coding practices, and share reusable learnings with the team
  • Apply Software Development Life Cycle tools, including authorized AI-assisted development and automation capabilities, to improve automation outcomes
  • Produce architecture and design artifacts for complex applications and ensure software implementation meets design constraints
  • Analyze large and varied datasets, creating visualizations and reports to support ongoing improvements to applications and systems
  • Identify underlying issues and patterns in data and use insights to improve coding practices and system architecture
Technologies:
  • AI
  • API
  • AWS
  • Cloud
  • Cassandra
  • Docker
  • GCP
  • Support
  • Java
  • Kubernetes
  • Machine Learning
  • NoSQL
  • PyTorch
  • Python
  • Security
  • TensorFlow
  • Web
  • microservices
  • DDD
  • Model Serving

More:

We are recruiting a Software Engineer III for our Commercial and Investment Banking team, where you will help build secure, stable, and scalable technology products as part of an agile team. J.P. Morgan Chase is a longstanding financial institution serving consumers, businesses, and public-sector and institutional clients, with operations spanning investment, commercial and consumer banking, payments, and asset management. Our Commercial & Investment Bank serves clients in more than 100 countries across banking, markets, securities services, and payments. We offer competitive compensation and, for eligible roles, incentive pay and benefits that may include health coverage, wellness centers, retirement savings, backup childcare, tuition reimbursement, mental health support, and financial coaching. We value diversity and inclusion, provide reasonable accommodations, and are an equal opportunity employer. This role is subject to applicable Section 19 requirements, including a review of criminal conviction history as described in the posting.

last updated 42 week of 2026

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Software Engineer, AI/ML & GPU Serving in Palo Alto, CA vacancy
  •  ...opportunity for you to take your software engineering career to the next...  ...enterprise-authorized AI coding assist tools...  ...experience, with emphasis on ML systems.Hands-on...  ...deploying or operating GPU workloads in Kubernetes...  ...environmentsExposure to ML model serving frameworks (e.g.,... 
    Suggested

    JP Morgan Chase

    Palo Alto, CA
    3 days ago
  • $120k - $195k

     ...responsible for scaling LinkedIn’s AI model training, feature engineering and serving with hundreds of billions...  ..., data infra, compute software, and hardware to harness the power of our GPU fleet with thousands of...  ...with hundreds of new ML models per quarter using thousands... 
    Suggested
    For contractors
    Work experience placement
    Work at office
    Flexible hours

    Linkedin

    Mountain View, CA
    2 days ago
  •  ...builds the world's largest AI chip, 56 times larger...  ...10 times faster than GPU-based hyperscale cloud...  ...Cerebras Wafer-Scale Engine.We are hiring a Software Engineer to...  ...and optimize our GPU serving stack, working across...  ...or another open-source ML systems project.Experience... 
    Suggested

    Cerebras Systems

    Sunnyvale, CA
    7 hours ago
  •  ...generation computing experiences—from AI and data centers, to PCs,...  ...:We're looking for a senior software engineer who combines deep systems...  ...someone who can shape software from GPU kernels through distributed...  ...CUTLASS, Thrust, CUB, NCCL), and ML framework cores such as... 
    Suggested
    Shift work

    AMD

    Santa Clara, CA
    4 days ago
  • $119.8k - $234.7k

     ...Less than 25%Profession: Software EngineeringDiscipline:...  ...builds the end-to-end AI stack and is core to Azure...  ...as a Service, Azure ML, Cognitive Services, and...  ...customers are better served. Within CoreAI, the Foundry...  ...a Principal Software Engineer - Responsible AI who is... 
    Suggested
    Ongoing contract
    Work at office
    Local area
    3 days per week

    Microsoft

    Mountain View, CA
    1 day ago
  • $147k - $210k

     ...models using advanced AI techniques to predict user...  ...of experience with software development in Python or...  ...statistical methods to solve engineering or scientific problems....  ...systems or specialized ML tools such as...  ...that can train on and serve on every query.Google's... 

    Google

    Mountain View, CA
    15 hours ago
  • $102.1k - $202.2k

     ...: Less than 25%Profession: Software EngineeringDiscipline: Software...  ...MicrosoftOverviewSoftware engineer role in Cloud + AI team, enabling secure,...  ...the inference infrastructure serving cutting-edge AI models at massive...  ...design, understanding of ML/AI fundamentals and... 
    Ongoing contract
    Local area
    3 days per week

    Microsoft

    Mountain View, CA
    3 days ago
  • $193.93k - $352.29k

     ...profound opportunity for AI to drive positive...  ...autonomously inside Nuro's own engineering organization, under the...  ....About You3+ years of software engineering experience...  ...helped.Experience with ML training or research...  ...Experience running inference serving, cost, or capacity at... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    2 days ago
  • $272k - $431.25k

    We are seeking a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to join our Hardware Infrastructure team. As an Engineer, you will have a pivotal role in enhancing efficiency for our researchers by implementing progressions throughout the entire stack... 
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $165.2k - $223.6k

     ...the right moment. As AI models outgrow any...  ...largest models can serve real users. That...  ...'re looking for an engineer to work at the frontier...  ...data-movement software that transfers KV cache...  ...interconnects, or GPU/accelerator systems...  ...is welcome.Prior AI/ML experience is not... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    1 day ago
  •  ...usher in this new era, we seek AI-native thinkers across...  ...price at scale across data engineering, analytics, and AI/ML workloads. Every day, new customers...  ...Migrations organization serves as the catalyst for this...  ...early products. OUR IDEAL SOFTWARE ENGINEER WILL HAVE: Strong... 

    Snowflake

    Menlo Park, CA
    2 days ago
  • $174k - $252k

     ...year of experience with software design and architecture...  ...integrating generative AI tools or LLM interfaces...  ...job Google's software engineers develop the next-generation...  ...with Machine Learning (ML) models. Information...  ...of the users we serve, creating a culture of... 
    Local area

    Google

    Mountain View, CA
    2 days ago
  • $182k - $242k

     ...Essential Cloud for AI™. Built for pioneers...  ...for high-performance GPU infrastructure across AI/ML, visual effects,...  ...inference. Our stack is engineered for speed, scale,...  ...of large-scale model serving—squeezing maximum throughput...  ..., GPU/accelerator software, or performance-... 
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    1 day ago
  •  ...Machine Learning Engineer LiveX AI is building the next generation...  ...train, fine-tune, and serve the generative models...  ...infrastructure on GPU clusters (PyTorch, DeepSpeed...  ...~3+ years of hands-on ML engineering experience...  ...pipelines. ~ Solid software engineering... 

    Livex AI

    Palo Alto, CA
    1 day ago
  •  ...profound opportunity for AI to drive positive...  ...About the Role As a software engineering intern, you will work closely...  ..., Onboard Systems, ML Infrastructure, Simulation...  ...: The Data Platform serves as a comprehensive management...  ...modalities (CPU, GPU, FPGA) etc. You have... 
    Internship
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $61k - $101k

     ...Requirements: Formal software engineering training or...  ...automation and CI workflows for ML codebases, including build...  ...systems across CPU/GPU utilization, memory, I/...  ...effective use of approved AI-assisted software development...  ...financial institution serving consumers, businesses,... 
    Full time

    J.P. Morgan

    Palo Alto, CA
    2 days ago
  • $145k - $170k

     ...profound opportunity for AI to drive positive...  ...the Role Our software team is growing,...  ...looking for talented engineers who are graduating...  ..., Onboard Systems, ML Infrastructure, Simulation...  ...The Data Platform serves as a comprehensive...  ...modalities (CPU, GPU, FPGA) etc. You... 
    Full time
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $174k - $252k

     ...5 years of experience with software development, applied ML, autonomous systems, and large...  ...the job Google's software engineers develop the next-generation...  ...high-velocity, autonomous AI systems. Google Ads is at...  ...representative of the users we serve, creating a culture of... 
    Shift work

    Google

    Mountain View, CA
    2 days ago
  •  ...builds Industrial AI for the world's leading...  ...state-of-the-art ML to large volumes of...  ..., fine-tuning, and serving them in large-scale...  ...a Senior/Staff AI Engineer, you will turn ML research...  ...AI Scientists, Software Engineers, and...  ...parallel training (multi-GPU/multi-node).... 
    Full time
    Shift work

    Gauss Labs

    Palo Alto, CA
    1 day ago
  •  ...Aug 25, 2026 Senior | Agentic AI & Applied ML US-based — CA preferred,...  ...billions of repair orders and serving live search at sub-second latency...  ...on topics including: engineering-diagram and technical-document...  ...similar), vLLM/TGI/TensorRT-LLM, GPU inference optimization... 
    Part time
    Remote work

    Atlassian

    Palo Alto, CA
    5 hours ago
  •  ...Description Accellor is an AI-native services firm...  ...advanced AI, data, and engineering capabilities. Our mission...  ...inference runtime, model serving, GPU infrastructure,...  ...years of experience in software engineering, systems architecture, ML infrastructure, distributed... 

    Accellor

    Mountain View, CA
    28 days ago
  • $147k - $210k

    Analyze ML workloads, identify compute and memory bandwidth bottlenecks...  ....Optimize auxiliary serving and distributed data pipelines...  ...issues at all levels of the ML software stack.Experience with ML compilers...  ...passes.Google's software engineers develop the next-generation technologies... 

    Google

    Mountain View, CA
    4 days ago
  •  ...inference provider focused on realtime AI for consumer-facing...  ...build first-party speech models, serve LLMs, and run the inference behind...  ...for a Founding AI Solutions Engineer who will sit at the intersection...  ...a plusDeep familiarity with AI/ML APIs — you understand how to work... 
    Full time
    Contract work
    Work at office
    Relocation

    Inworld AI

    Mountain View, CA
    2 days ago
  • $174k - $252k

     ...evaluate training and serving metrics for next-generation...  ....Develop and integrate AI-agentic platforms and...  ...and optimization engines (e.g., Mixed-Integer Programming...  ...experience.5 years of software engineering (e.g., C++...  ...) across modern ML frameworks (e.g., JAX,... 

    Google

    Sunnyvale, CA
    1 day ago
  • $147k - $210k

     ...on offline/omni shopping ads by refining ML models, auction, ads triggering (query...  ...technologies.Experience in analytics.Google's software engineers develop the next-generation technologies...  ...key components of the Google Ads serving system. You will get deep insights into... 

    Google

    Mountain View, CA
    3 days ago
  • $174k - $252k

     ...rights-protected content across all GenAI serving surfaces.Optimize large-scale Retrieval-...  ...leadership and mentorship to other engineers within a rapidly growing organization, guiding...  ...experience.5 years of experience with software development in C++ and JavaExperience in... 

    Google

    Mountain View, CA
    2 days ago
  •  ...ourselves — real-time GPU serving, high-throughput batch...  ...infrastructure for Generative AI at DoorDash, with a...  ...serving and inference engines, fine-tuning and training...  .... Partner closely with ML engineers, product...  ...industry experience in software engineering Strong backend... 

    DoorDash, Inc.

    Sunnyvale, CA
    2 days ago
  • $207k - $300k

     ...co-evolve ads, leverage AI to maintain business...  ...advertisers.Leverage AI/ML advancements, including...  ...systems for deploying and serving AI/ML models...  ...years of experience in software development.8 years of...  ...practices.Google's software engineers develop the next-generation... 

    Google

    Mountain View, CA
    4 days ago
  • $193.3k - $261.5k

     ...builds AWS Neuron, the software development kit used to...  ...Inferentia and Trainium ML accelerators. This...  ...software boundary, our engineers build systematic infrastructure...  ...of what's possible in AI acceleration.As part of...  .../offline inference serving with vLLM, SGLang, TensorRT... 
    Work experience placement
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    15 hours ago
  • $207k - $300k

     ...maintain LLM training and serving benchmarks; use...  ..., drive XLA:GPU/Triton performance...  ...identify bottlenecks and engineer scalable...  ...years of experience in software development.5 years...  ...of experience with ML design and ML infrastructure...  ...deployment on AI accelerators.Preferred... 

    Google

    Sunnyvale, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Software Engineer, AI/ML & GPU Serving. Be the first to apply!