Senior Staff AI Software Engineer
San Jose, California, United States
USD 189K-301K Senior-level Full Time
Tasks
- Analyze and optimize LLM and agentic AI workloads
- Collaborate with hardware teams on memory architecture and acceleration
- Define best practices and mentor across software hardware co-design
- Develop optimize and benchmark inference and serving solutions
- Lead software and hardware co-design for AI inference performance
- Profile model execution for memory bottlenecks
Perks/Benefits
- 401k
- Adoption support
- Charitable giving match
- Confidential Therapy Sessions
- Fertility care stipend
- Flexible work environment
- Holidays
- Medical travel support
- Medical/Dental/Vision insurance
- Onsite cafe
- Onsite gym
- Paid time off
- Sick leave
- Virtual classes
- Virtual vet care
Skills/Tech-stack
GitHub | High Bandwidth | High Bandwidth Memory | Jira | Linux | PyTorch | Transformer | VLLM
Education
Related jobs
-
Software Engineer, Systems ML Engineering USD 170K-251KAlerting | Benchmarking | C++ | CUDA | DashboardingSenior-level Full TimeSunnyvale, CA | Bellevue, WA | …2h ago
-
Research Software Engineer, Multimodal AI USD 174K-253KAgent Orchestration | Audio | C++ | Few-Shot Learning | Few-shotMid-level Full TimeSan Jose, CA, USA2h ago
-
Senior Software Engineer, AI/ML, Geo and Gemini App USD 174K-253KA/B | A/B Testing | B testing | C++ | Data AnalysisSenior-level Full TimeNew York, NY, USA2h ago
-
Senior Software Engineer, AI/ML, Ads Training USD 174K-253KC++ | Distributed Systems | GPU | JAX | Machine LearningSenior-level Full TimeMountain View, CA, USA3h ago
-
Software Engineer, Product USD 208K-269KAgentic Frameworks | Anthropic | Claude Code | Cursor | DocumentationDental insurance | Health insurance | Paid time off | Retirement plan | Vision insuranceMid-level Full TimeSan Francisco, CA7h ago
-
C++ | Containers | Distributed Systems | Distributed Training | Distributed inferenceSenior-level Full TimeCupertino8h ago
-
Machine Learning Engineer, Sensor Pipelines USD 175K-215KC++ | Computer Vision | JAX | Machine Learning | PyTorchHybrid work scheduleMid-level Full TimeMountain View, CA, USA14h ago
-
Sr. Embedded Software/Firmware Engineer USD 180K-240KARM | ASPICE | AUTOSAR | C# | C++Competitive compensation | Dental insurance | Flexible hours | Medical insurance | Pre-IPO stock optionsSenior-level Full TimeUnited States - Remote R14h ago
-
Data Engineer USD 112K-160KAWS | AWS S3 | Airflow | Azure | CI/CD401k match | Family leave | Medical, dental, and vision insurance | PTO | Paid HolidaysMid-level Full TimeSanta Monica, CA, United States14h ago
-
Senior Numerical Optimization Engineer USD 191K-253KAPIs | C++ | CI/CD | CPLEX | Cloud ComputingComprehensive benefits | Health insurance | Paid on-call supportSenior-level Full TimeCosta Mesa, California, United States15h ago
-
Sr. Software Engineer, AI Infrastructure USD 144K-236KApache Beam | Apache Flink | Apache Spark | C# | C++Career development | Hybrid work | Stock grantsSenior-level Full TimeSunnyvale, CA, United States15h ago
-
Software Engineer, ML Platform (ML Training) USD 175K-265KAWS | Computer Vision | DeepSpeed | Distributed Training | JAXMid-level Full TimeFoster City, CA15h ago
-
Senior Data Engineer USD 180K-210KAirflow | Anomaly Detection | BigQuery | CI/CD | Cloud platform401k employer match | Annual Company Holidays | Catered lunches | Company retreat | Comprehensive health coverageSenior-level Full TimeNew York City15h ago
-
Sr. Engineer, Firmware & Embedded Software USD 79K-158KAgile | Azure DevOps | BDM | Bare Metal | BootloaderOccasional travel for onboarding | Remote work within the United StatesSenior-level Full TimeFlorida, United States R16h ago
-
Senior Data Engineer - Digital Pathology (Hybrid) USD 98K-163KAPI Integration | Agile | Apache Airflow | Apache Spark | Batch ProcessingSenior-level Full TimeRochester, MN, United States R17h ago
-
Machine Learning Research Engineer, App Store Search USD 185K-313KA/B | A/B Testing | Apache Spark | B testing | Fine TuningMid-level Full TimeSeattle17h ago
-
Principal Machine Learning Engineer USD 126K-264KCode review | Continuous Delivery | Continuous integration | Data Preprocessing | Data Privacy401k match | Adoption Assistance | Dental insurance | Health insurance | Life insuranceSenior-level Full TimeSeattle, WA, United States17h ago
-
Machine Learning Engineer - Vision USD 160K-250KActive Learning | Annotation tools | CAD/CAM | Computer Vision | Image Segmentation401k | Dental insurance | Equity | Flexible vacation | Health insuranceSenior-level Full TimeLos Angeles, CA17h ago
-
Machine Learning Engineer - Multimodal USD 160K-250KActive Learning | Computer Vision | Data Augmentation | Data loaders | Evaluation Frameworks401k | Equity | Flexible vacation policy | Medical/dental/vision/life insurance | Relocation supportSenior-level Full TimeLos Angeles, CA17h ago
-
Machine Learning Engineer - LLMs USD 160K-250KActive Learning | DPO | Data Augmentation | Data loaders | Document Classification401k | Equity | Flexible vacation policy | Medical/dental/vision/life insurance | Relocation supportSenior-level Full TimeLos Angeles, CA17h ago
-
Cheminformatics | Cloud Computing | Data Analysis | Data Visualization | Deep learningSenior-level Full TimeRedwood City, California, United States19h ago
-
AI Developer USD 99K-198KAWS | Agentic AI | Azure | CI/CD | Data LakesHybrid work schedule | Leadership development | Professional developmentMid-level Full TimeWashington, DC19h ago
-
Software Engineer USD 85K-100KBiometrics | Computer Vision | Deep learning | Facial recognition | Image Geolocation401k match | Dental insurance | Disability insurance | Health insurance | Life insuranceMid-level Full TimeSomerville, Massachusetts, United States19h ago
-
Database Developer (Contractor) - MNSITE-3762 USD 100K-125KAgile | Azure DevOps | Data Cleansing | Data Migration | Data ModelingFull time contract consulting | On-call support rotation | Remote work | Virtual interviews via Microsoft TeamsMid-level ContractSaint Paul, MN, United States20h ago
-
Senior Machine Learning Engineer, Safety USD 216K-303KData Preprocessing | Deep learning | Distributed Training | Embeddings | Hugging FaceSenior-level Full TimeRemote - United States R21h ago