Staff Python / PyTorch Developer — Frontend Inference Compiler – Dubai
Europe; Remote, California, United States; UAE
R
USD 175K-260K (estimate) Senior-level Full Time
Tasks
- Analyze new generative AI models for compilation impact
- Build frontend compiler infrastructure to produce intermediate representation
- Collaborate with teams on feature implementation
- Develop model definition framework for large language models
- Extend PyTorch FX TorchScript TorchDynamo tooling for graph capture transformation analysis
- Optimize PyTorch based graph transformations for inference platform
- Research model optimization methods for faster inference
Perks/Benefits
Skills/Tech-stack
C++ | Compiler development | Computational graphs | Generative AI | Graph Capture | Graph Transformation | Graph optimization | HuggingFace Transformers | Intermediate representation | Language Models | Large Language Models | ML Graph Optimization | MLIR | Model Tracing | Multi Target Compilation | Numerical precision | Open Source | Operator Lowering | PyTorch | PyTorch FX | Python | Quantization | Runtime scheduling | Tensor operations | TorchDynamo | TorchScript | Transformer Architecture
Education
N/A
Roles
Developer | Engineer | Learning Engineer | Machine Learning Engineer | Python | Python Developer
Related jobs
-
Machine Learning Engineer USD 128K-214KAWS | Agile | Azure | Cloud platform | GitHealth insurance | Holiday pay | Learning and development | Life insurance | Long-term disabilityMid-level Full TimeUSA-Remote Work R6h ago
-
Entry-level Full TimeUnited States - Remote R19h ago
-
CI/CD | Docker | Drift Detection | Embeddings | Experiment trackingMentorship | Remote workSenior-level Full TimeUnited States - Remote R19h ago
-
Machine Learning Engineer II USD 131K-184KAzure | Batch inference | Data Pipelines | Databricks | Distributed SystemsContinuous learning | Flexible ways of working | Growth mindset cultureMid-level Full TimeUSA TX Houston Hybrid, United States R19h ago
-
Backend Software Engineer (GenAI) USD 104K-180KAPI Development | AWS | Amazon Bedrock | Amazon SageMaker | CI/CDMid-level Full TimeUnited States Remote, United States R19h ago
-
C plus plus | C# | CAD | Dynamics | FDA Compliance401k | Company holidays | Dental insurance | Health insurance | Paid maternity/paternity leaveSenior-level Full TimeLos Angeles, California R19h ago
-
Senior Data Engineer USD 117K-162KAWS | Azure | BigQuery | DBT | Data Architecture401k | Annual wellness stipend | Cell phone reimbursement | Coaches and therapists access | Collective Pause DaysSenior-level Full TimeRemote - US R21h ago
-
Embedded Software Engineer II USD 115K-140KBash | C plus plus | C# | CI/CD | D-busERGs | Family Caregiver Support | Flexible PTO | HSA match | Health benefitsMid-level Full TimeRemote - USA R22h ago
-
AI Engineer USD 131K-185KAnthropic API | Apps Script | Autogen | Cloud deployment | CrewAIAsync first collaboration | Conversion to employment based on performance | Direct access to leadership | Fast feedback loops | Fully remoteMid-level Full TimeUnited R22h ago
-
Senior Solution Engineer USD 165K-216KAnalytics | Cloud Computing | Data Architecture | Data Lake | Data WarehouseSenior-level Full TimeUS-CA-Bay Area-Remote R23h ago
-
Senior Software Engineer USD 140K-185KAWS | Automated testing | Azure | C++ | Git401K company matching | Dental insurance | Dependent care benefits | Flexible spending account | Health insuranceSenior-level Full TimeBoulder, CO R1d ago
-
Software / Computer Science Intern USD 42K-50KData Parsing | Data Querying | Data Storage | Data pipeline | DebuggingCollaborative team activities | Hybrid work arrangement | Mentorship | Occasional local travel | Professional developmentEntry-level InternshipMonroeville, PA R1d ago
-
Machine Learning Engineer USD 150K-215KData Augmentation | Deep learning | Isaac | Loss Functions | Medical ImagingMid-level Full TimeSan Francisco (hybrid) R1d ago
-
AI Engineer (Latam, Remote) USD 70K-110KAPI Integration | Authentication | Claude | Database Management | LLM integrationCollaboration with U S based teams | Fully remote | High ownership and autonomy | Part-time flexibilitySenior-level Part TimeFlorida, Aventura, United States of America R1d ago
-
Software Engineer II - Model Platform USD 149K-214KAWS | Azure | Cloud Computing | Data Pipelines | Distributed SystemsMid-level Full TimeRemote - USA R1d ago
-
Data Platform Engineer III USD 135K-160KAPIs | AWS Lambda | Agile | Amazon RDS | Amazon S3401k employer match | Dental insurance | ESPP | Flexible spending account | Health insuranceSenior-level Full TimeRemote, United States R1d ago
-
Senior Software Engineer (Typescript / FrontEnd) - AI/ML USD 141K-232KAPI Design | AWS | Azure | Cloud platform | Google CloudFlexible time off | Flexible work environment | Global gatherings | Healthcare | Home office setupSenior-level Full TimeUnited States (remote) R1d ago
-
Data Scientist / AI/ML Engineer (Imagery) VAWFH 1652 USD 153K-207KAccuracy | Computer Vision | Containerization | Data Cleansing | Data PreprocessingSenior-level Full TimeReston, VA R1d ago
-
Senior Machine Learning Ops Engineer USD 150K-173KAWS | Airflow | Bash | Batch inference | CI/CDEmployee mentorship program | Leadership programsSenior-level Full TimeUnited States R1d ago
-
Forward Deployed Machine Learning Engineer USD 180K-300KAPI Design | Cloud Computing | Deep learning | Diffusion Models | Fine TuningIn-person collaboration days | Remote work flexibility | Travel cost coverageSenior-level Full TimeSan Francisco (USA) R1d ago
-
Senior Software Engineer - Platform & MLOps USD 152K-230KAWS | Azure | CI/CD | Datadog | DockerDiscretionary incentive plan | Flexible work policy | Learning and development access | Medical benefitsSenior-level Full TimeSeattle, Washington, United States - Remote R1d ago
-
C# | MATLAB | NumPy | Pandas | PythonEnglish language support | Flexible schedule | Part-time project-based workSenior-level FreelanceUnited States - Remote R1d ago
-
APIs | Agent architecture | Embeddings | Inference Serving | LLMDirect technical influence | Early stage equity upside | Fast Moving Engineering Culture | High technical autonomy | Remote workMid-level Full TimeSan Francisco, CA; Onsite R1d ago
-
Data Engineer (Remote) USD 97K-130KAWS | Agile | Azure | Azure DevOps | CI/CDOff hours production support rotation | Remote workSenior-level Full Time1 First American Way, Santa Ana, … R1d ago
-
Principal Data Engineer - Parametric USD 115K-225KAgile | Amazon Web Services | Athena | Containerization | Data GovernanceSenior-level Full TimeSeattle WA 800, United States R1d ago