AI and ML Infra Software Engineer, GPU Clusters - New College Grad 2026
Tasks
- Collaborate with AI and ML research teams to define infrastructure needs
- Collaborate with researchers data engineers and DevOps professionals to build AI ML infrastructure ecosystem
- Define AI researcher efficiency metrics and track measurable results
- Ensure high availability scalability and efficient utilization
- Implement AI and ML infrastructure improvements based on latest technologies
- Monitor infrastructure performance and optimize resource utilization
- Resolve infrastructure gaps for GPU cluster research
Perks/Benefits
Skills/Tech-stack
AWS | Amazon EFA | Azure | Bash | BeeGFS | Cloud Computing | DDP | Data Processing | Data-parallel | Distributed Data Parallel | Distributed Training | Distributed data | Docker | Enroot | FSDP | Fully Sharded Data Parallel | GCP | GPFS | GPU | Go | High Performance | High speed | High-Performance Computing | High-Speed Networking | Inference Pipelines | Infiniband | JAX | Kubernetes | Lustre | Model Training | NEMO | Parallel Computing | Performance Computing | PyTorch | Python | RoCE | Scheduling and orchestration | Slurm | Storage Systems
Education
Regions
Countries
States
Related jobs
-
Senior Software Engineer, Deep Learning - MLIR TRT USD 184K-287KC# | C++ | CUDA | Computer Vision | Deep learningSenior-level Full TimeUS, CA, Santa Clara1d ago
-
C# | C++ | DRAM | Distributed Systems | GPUSenior-level Full TimeUS, CA, Santa Clara R1d ago
-
AI infrastructure | C++ | CUDA | Data Preprocessing | Data loadingBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara1d ago
-
Senior Deep Learning Frameworks Sustaining Engineer USD 152K-218KBazel | C# | C++ | CI/CD | CUDAEquity | Health insurance | Paid time offSenior-level Full TimeUS, CA, Santa Clara1d ago
-
Senior Machine Learning Data Curation Engineer USD 174K-295KAWS | Big Data | BigQuery | Data Annotation | Data GovernanceComputational resources | Dinners | Fun activities | Lunches | SnacksSenior-level Full TimeSanta Clara, CA2d ago
-
BIOS | BMC | Bash | C# | C++Senior-level Full TimeUS, CA, Santa Clara2d ago
-
Distinguished Engineer, Storage – AI Cloud USD 320K-488KAPI Design | Availability | BeeGFS | Block Storage | By DesignSenior-level Full TimeUS, CA, Santa Clara2d ago
-
Principal Deep Learning Algorithm Engineer USD 272K-431KC++ | CUDA | Computer Architecture | Deep learning | GPU ProgrammingBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R3d ago
-
Supply Chain Data Engineer USD 152K-230KAWS | Apache Airflow | Azure | BigQuery | DBTBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara3d ago
-
ADA | ARM | ARMv8 | BSP | Bring-upBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara3d ago
-
Senior Cloud Software Engineer, DGXC Data Services USD 152K-241KAPI Design | AWS | Algorithms | C++ | Data StructuresBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R3d ago
-
AI readiness | AWS | Access Management | Architecture | AsanaCareer growth opportunities | Customer facing exposure | Remote workSenior-level Full TimeHonduras - Tegucigalpa3d ago
-
C++ | CUDA | Containerization | Distributed Systems | DockerSenior-level Full TimeUS, CA, Santa Clara R4d ago
-
Principal Machine Learning Engineer USD 157K-254KAWS | Apache Flink | Azure | CI/CD | CloudFormationFlexible work schedule | In-office work flexibilitySenior-level Full TimeSanta Clara, CA4d ago
-
Lead Generative AI Analyst - English US USD 122K-186KData Analysis | Data labeling | Documentation | Machine Learning | Process ImprovementSenior-level Full TimeSanta Clara, CA4d ago
-
C++ | Computer Vision | Curriculum learning | Data pipeline | Deep learningSenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Senior Deep Learning Framework Communications Engineer USD 152K-241KC++ | CUDA | Communication overlap | Compiler optimization | Compute Communication OverlapSenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Performance Engineer, Deep Learning and HPC USD 116K-189KAWS | Amber | Azure | Compilers | Data AnalysisSenior-level Full TimeUS, CA, Santa Clara5d ago
-
C++ | CUDA | Distributed inference | GPU Architecture | Graph CompilerBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
3D Computer Vision | 3D representations | Active Learning | Auto-labeling | BEVBenefits | EquitySenior-level Full TimeUS, CA, Santa Clara5d ago
-
Senior System Software Engineer – Dynamo Tools USD 152K-241KAWS | Automation | Azure | Cloud platform | ConcurrencyComprehensive benefits package | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Senior Software Engineer, AI Storage USD 184K-287KAlgorithms | Bash | C# | C++ | CUDABenefits | EquitySenior-level Full TimeUS, CA, Santa Clara R5d ago
-
Bash | Bootstrap | CSI | CSS3 | Container StorageSenior-level Full TimeUS, CA, Santa Clara R6d ago
-
ARM Cortex | ARM Cortex-M | BMC | BMC firmware | BashSenior-level Full TimeUS, CA, Santa Clara7d ago
-
Senior Applied AI Engineer USD 152K-287KAPI Design | Agentic Workflows | Architecture | Automation | CI/CDEquity | Health benefitsSenior-level Full TimeUS, CA, Santa Clara8d ago