Member of ML Technical Staff
Tasks
- Build and optimize pre training and post training pipelines
- Build research infrastructure
- Collaborate on software hardware co design for throughput
- Design and implement training techniques
- Develop distributed training across GPU clusters
- Explore model architectures training methodologies post training techniques
- Implement model parallel training strategies
- Improve training and inference kernels
- Improve training efficiency throughput stability scalability
- Optimize GPU workloads
- Research train improve large language models
- Run experiments analyze results iterate quickly
Perks/Benefits
Skills/Tech-stack
CUDA | Compiler optimization | Data parallelism | Distributed Training | GPU Profiling | High Performance | High-Performance Computing | JAX | Performance Computing | Pipeline parallelism | PyTorch | Python | Sequence parallelism | Tensor Parallelism | Triton
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
Regions
Countries
States
Related jobs
-
Machine Learning Engineer, Memory USD 125K-194KAgile | Computer Vision | Foundation Models | Horizon Modeling | Language Processing401k match | Coaching | Dental insurance | Life insurance | Long-term disabilityMid-level Full TimeCary,North Carolina,United States5h ago
-
A/AI Machine Learning Engineering Stf - E4 USD 135K-251KAI | AI RMF | Adversarial Machine Learning | Automation Testing | C#401k match | Dental insurance | EAP | Education assistance | Employee assistance programSenior-level Full TimeShelton, CT, US5h ago
-
PIIM Data Engineer USD 119K-182KAPI Integration | APIs | Azure Databricks | Data Governance | Data LakeFull Time Office Project Telework | Relocation assistanceMid-level Full TimeNew Albany, OH, US5h ago
-
Software Engineer, Data Platform Team USD 152K-179KApache Airflow | Data APIs | Data Catalog | Data Lineage | Data Modeling401k matching | Medical, dental, and vision | Quarterly in person surges | Remote-first work environmentMid-level Full TimeRemote - USA R6h ago
-
Senior Machine Learning Engineer USD 229K-360KAB Testing | APIs | Airflow | Algorithm Development | Amazon S3Disability benefits | Health insurance | Life insurance | Paid Equity Awards | Paid time offSenior-level Full TimeNew York, New York R6h ago
-
Senior Machine Learning Engineer USD 229K-360KA/B | A/B Testing | Amazon S3 | Amazon SageMaker | Apache AirflowCommuter benefits | Disability benefits | Equity awards | Financial wellness support | Health insuranceSenior-level Full TimeSan Jose, California R6h ago
-
Senior AI Engineer – AI Platform USD 140K-175KAWS | Drift Detection | Embeddings | Evaluation Pipelines | Function CallingDaily group classes | Dry cleaning | Free onsite fitness center | Free shuttle transportation | Gourmet cafeteriaSenior-level Full TimeEnglewood Cliffs, NEW JERSEY, United States R6h ago
-
Senior ML Engineer – AI Platform USD 140K-175KCNN | Deep learning | Embeddings | Human-in-the-loop | Information ExtractionDaily group classes | Dry cleaning | Free onsite fitness center | Free shuttle transportation | Gourmet cafeteriaSenior-level Full TimeEnglewood Cliffs, NEW JERSEY, United States6h ago
-
AI Engineer USD 220K-260KAPI Development | Agent Orchestration | Data Generation | Debugging | Fine Tuning401k matching | Dental insurance | Fertility and family building stipend | Flexible PTO | Health insuranceMid-level Full TimeSan Francisco R8h ago
-
Forward Deployed Engineer, Agentic AI USD 180K-180KAI Security | AWS Bedrock | Access Control | Agent Development | Anthropic ClaudeMid-level Full TimeUS - Remote R8h ago
-
Staff Machine Learning Engineer - Ads USD 164K-282KAWS | Airflow | Apache Spark | BigQuery | Cloud Computing401k plan with company match | AD&D insurance | Disability insurance | Electric Car Charging Station | Employee assistance programSenior-level Full TimeMountain View, USA9h ago
-
Agents | Golang | Information Retrieval | Language Models | Language Processing401k company match | ESPP | Family leave programs | Flexible spending accounts | Flexible time offSenior-level Full TimeMountain View, CALIFORNIA, United States R10h ago
-
Director, AI & Agentic Engineering CAD 200KAI Foundry | AWS Bedrock | Agent Orchestration | Agentic Systems | Artificial IntelligenceHybrid workExecutive-level Full Time135 W 26th Street, New York, … R10h ago
-
ASIC | Agile | Automation frameworks | BDD | C#Senior-level Full TimeCupertino, California, USA10h ago
-
Staff Machine Learning Engineer USD 176K-308KAlgorithms | Code Tuning | Design Patterns | High Performance | High-Performance Computing401k match | ESPP | Family leave programs | Flexible spending accounts | Flexible time awaySenior-level Full TimeSanta Clara, California, United States R10h ago
-
GTM AI Engineer USD 112K-140KAPI | Chunking | Compute cost management | Context windows | Cost Management401k matching | Dental insurance | Health insurance | Life insurance | Long-term disabilityMid-level Full TimeUnited States - Remote R11h ago
-
Mid-level Full TimeSan Francisco R11h ago
-
Founding Data & Analytics Engineer (NYC) USD 150K-300KDBT | FHIR | GraphQL | Language Processing | Machine LearningFlexible PTO | Health savings account | HealthAdvocate | Lunch and dinner | Medical/Dental/Vision insuranceSenior-level Full TimeNew York City (Soho)12h ago
-
Senior Backend Engineer, Data Ingestion USD 145K-230KAWS | Amazon SQS | Celery | Data Pipelines | Database Query401k | Commuter benefits | Daily catered lunch | Dental insurance | Flexible paid time offSenior-level Full TimeRemote R12h ago
-
AI/ML Software Engineer USD 80K-80KAWS | Angular | Azure | CI/CD | Data Pipelines401k matching | Flexible PTO | Life insurance | Medical/Dental/Vision insurance | Paid parental leaveMid-level Full TimeCharlotte, NC13h ago
-
Sr Data Engineer USD 109K-156KAPI Management | Access Control | Airflow | Amazon Redshift | Apache SparkSenior-level Full TimeRemote, United States R13h ago
-
Sr Engineer, Data Engineering USD 32K-46KAmazon Web Services | Apache Spark | Azure | Data Lake | Data WarehouseSenior-level Full TimeKnoxville, TN, United States13h ago
-
Senior Data Engineer, AI Platform USD 134K-168KAWS | Amazon Redshift | Apache Airflow | Apache Iceberg | Apache Kafka401k match | Insurance Coverage for Dependents | Learning and wellness stipend | Paid parental leave | Remote-firstSenior-level Full TimeUnited States13h ago
-
Embedded Software Engineer, Onboard Platform USD 100K-220KAutomotive networking | C++ | CAN | Communication Protocols | DO-178CSenior-level Full TimeSan Francisco, CA R13h ago
-
Operations Engineer, Applied AI (Spaceport Facilities) USD 125K-220KAI integration | Agent systems | Enterprise AI | Enterprise AI integration | LLMOnsite work | Travel opportunitiesEntry-level Full TimeStarbase, TX R13h ago