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
  • $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
    3 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
    2 days 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
    18 hours 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... 
    Suggested

    Google

    Mountain View, CA
    1 day 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
    1 day 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
    3 days 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
    4 days 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
    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
    2 days 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
    2 days 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
    18 hours 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
    18 hours 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
    3 days 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
    29 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
    3 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
    2 days 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
    4 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
    3 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
    18 hours 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
    1 day ago
  • $207k - $300k

     ...integration with high-throughput Ad serving systems.Explore, implement,...  ....8 years of experience in software development.5 years of experience...  ...systems.Google's software engineers develop the next-generation technologies...  ...Ads is at the forefront of AI innovation, applying cutting-... 

    Google

    Mountain View, CA
    4 days 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
    18 hours ago
  • $207k - $300k

     ...systems, while mentoring engineers in measurement best practices...  ...8 years of experience in software development.5 years of...  ...of experience leading ML design and optimizing ML...  ...learning and data science to serve as a technical lead. You will evaluate AI Overviews for factual... 

    Google

    Mountain View, 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
    18 hours ago
  • $136.3k - $199.9k

     ...teams of physicists, engineers, data scientists...  ...their algorithms. AI, including several...  ...Computing - HPC (including GPU), Machine Learning,...  ...hardware and software technologies that enable...  ..., and model serving systems.Research &...  ...server management.AI/ML Integration: Understanding... 
    Minimum wage
    Full time
    Work experience placement
    Flexible hours

    KLA-Tencor

    Milpitas, CA
    1 day ago
  • $153k - $176k

     ...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... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    a month ago
  • $145k - $200k

     ...the world’s leading software for data-driven decisions...  ...We are a software engineering team with expertise in enabling ML models in production. We deploy AI models to run in...  ...often with constrained GPU resources and...  ...Python and GoModel serving engines for GPU-accelerated... 
    Full time
    Work experience placement
    Work at office
    Remote work
    Work from home
    Relocation package

    Palantir Technologies

    Palo Alto, CA
    2 days ago
  • $137.2k - $206.6k

     ...We're looking for a Staff Software Engineer with broad software experience and a strong AI focus to help bring production...  ...prototype, and deploy AI/ML and computer vision...  ...environmentsExperience operating GPU compute for AI workloads — model serving, inference optimization or... 
    Full time
    Temporary work

    Joby Aviation

    San Carlos, CA
    2 days ago
  • $207k - $300k

     ...infrastructure (training and serving) to support new modeling...  ...testing, and launching software products.5 years of experience leading ML design and optimizing ML...  ...’s degree or PhD in Engineering, Computer Science, or a...  ...is at the forefront of AI innovation, applying cutting... 
    Immediate start

    Google

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

     ...at the right moment. As AI models outgrow any single...  ...'s largest models can serve real users. That network...  ...builds.We're looking for an engineer to work at the frontier...  ...low-level data-movement software across accelerators,...  ...experience is a plus.Prior AI/ML experience is welcome... 
    Internship
    Local area
    Flexible hours

    Amazon

    Cupertino, 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!