AI Inference Performance Engineer - New College Grad 2026
US, CA, Santa Clara, United States
USD 124K-241K Senior-level Full Time
Tasks
- Apply roofline analysis and profiling
- Architect execution from single GPU to rack scale clusters
- Benchmark serving performance at scale
- Contribute to TensorRT LLM vLLM and SGLang
- Drive end to end inference optimization pipeline
- Implement memory management optimization
- Implement quantization optimization
- Implement scheduling optimization
- Lead cross functional execution on benchmark timelines
- Manage cluster GPU performance
- Optimize distributed inference across frameworks
- Profile CUDA kernels and framework bottlenecks
Perks/Benefits
Skills/Tech-stack
Attention | Batching | C++ | CUDA | CUDA kernels | Cutlass | Deep learning | Deep learning inference | Distributed inference | Graph lowering | JAX | KV cache | Kubernetes | MPI | Memory Management | NCCL | Operator fusion | Profiling | PyTorch | Python | Quantization | Roofline analysis | Scheduling | Speculative decoding | Torch compile | Triton
Education
Regions
Countries
States
Cities
Related jobs
-
Featured Feat. Applied AI Engineer - Bay Area USD 211K-263KArtificial Intelligence | C plus plus | C# | Embeddings | Feature Engineering401k | Comprehensive health and wellness benefits | Learning and development opportunities | Unlimited time offMid-level Full TimeHQ (San Francisco)24d ago
-
Senior Data Engineer (Remote) USD 155KAgile | Apache Spark | BigQuery | Cassandra | Data Governance401k match | Dental insurance | Employee assistance program | Employee stock purchase plan | Flexible scheduleSenior-level Full TimeWork From Home, United States R7h ago
-
Senior AI Operations Engineer USD 170K-180KAI infrastructure | Azure | CI/CD | Cloud infrastructure | Container Engine for Kubernetes401k match | Employee assistance program | Employee stock purchase plan | Flexible schedule | Flexible spending accountSenior-level Full TimeWork From Home, United States R7h ago
-
API Development | Airflow | Automated retraining | CI/CD | Cloud PlatformsEquityMid-level Full TimeNaples, United States9h ago
-
Adversarial Machine Learning | Anomaly Detection | Cloud Security | Machine Learning | PythonSecurity clearance premiumsMid-level Full TimeNaples, United States9h ago
-
API Design | AWS | AWS Cloud | AWS Cloud Development Kit | AWS cloud developmentSenior-level ContractGlendale, United States10h ago
-
Mid-level Full TimeUS-Kansas-Wichita10h ago
-
Senior-level Full TimeCincinnati, OH, United States11h ago
-
Delivery Senior Consultant, Data Engineering and Gen AI USD 119K-208K.NET | AWS | Agentic AI | Agile | AngularSenior-level Full TimeGilbert, Arizona, United States; Lake Mary, …11h ago
-
Software Engineer/Researcher, AI-Native Database Systems USD 156K-387KC++ | Database Architecture | Distributed Systems | Indexing | Information RetrievalSenior-level Full TimeSan Jose, California, United States11h ago
-
Agent architecture | Backend Development | Document ingestion | Frontend Development | IndexingCross-functional collaboration | Hands-on experience | MentorshipEntry-level InternshipSan Jose, California, United States11h ago
-
Apache Flink | Apache Spark | Automation | C++ | Cause analysisSenior-level Full TimeSan Jose, California, United States11h ago
-
Research Engineer / Scientist - Storage for LLM USD 156K-387KAttention Mechanisms | CUDA | Caching | Distributed Systems | Eviction policiesCompetitive compensation | Conference attendance | Generous research resources | Innovation-driven culture | Open source contributionsEntry-level Full TimeSan Jose, California, United States11h ago
-
Agentic data | Apache Hive | Apache Spark | Coding Data | Data CurationSenior-level Full TimeMenlo Park, CA12h ago
-
Cloud Data and AI Engineer, Professional Services USD 127K-183KBigtable | C++ | Cloud Databases | Cloud SQL | Cloud platformTravel up to 30%Mid-level Full TimeReston, VA, USA12h ago
-
Senior Software Engineer, AI/ML, Google Cloud AI USD 174K-253KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeKirkland, WA, USA; Sunnyvale, CA, USA12h ago
-
Customer Engineer II, Applied AI, Google Cloud USD 148K-216KC++ | Cloud Architecture | Conversational AI | Document AI | Generative AITravel opportunitiesSenior-level Full TimeSunnyvale, CA, USA; Chicago, IL, USA12h ago
-
Software Engineer III, AI/ML GenAI, Google Ads USD 147K-211KC plus plus | Data Processing | Debugging | Generative AI | Language ProcessingSenior-level Full TimeMountain View, CA, USA12h ago
-
Customer Engineer, AI/ML, Startups, Google Cloud USD 104K-151KAI Studio | Agent systems | Agentic Workflows | Artificial Intelligence | AutogenMid-level Full TimeAustin, TX, USA12h ago
-
Research Engineer, Benchmarking, Robotics, DeepMind USD 147K-211KBenchmarking | Data Pipelines | Deep learning | Docker | Embedded deploymentMid-level Full TimeMountain View, CA, USA12h ago
-
Algorithms | Android | C# | C++ | Data StructuresMid-level Full TimeSan Jose, CA, USA; Waterloo, ON, …12h ago
-
Senior Staff Software Engineer, Dataproc, Control Plane USD 262K-365KC++ | Cloud APIs | Cloud Computing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA12h ago
-
Senior Software Engineer, Embedded Systems/Firmware, Platforms Infrastructure Engineering USD 174K-253KC# | C++ | Code review | Debugging | Embedded operating systemsSenior-level Full TimeSunnyvale, CA, USA12h ago
-
Software Engineer USD 149K-211KAlgorithms | C# | C++ | Cause analysis | Code reviewBonus | Equity | Health benefits | Hybrid scheduleMid-level Full TimeMountain View, CA, USA R12h ago
-
Software Engineer USD 149K-211KAlgorithms | Angular | C# | C++ | Cause analysisBonus | Equity | Health insurance | Paid time off | Professional developmentMid-level Full TimeMountain View, CA, USA12h ago