Staff Python / PyTorch Developer — Frontend Inference Compiler – Dubai
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
-
Statistics & Python Expert - Freelance AI Trainer USD 146K-146KCombinatorics | Graph theory | MATLAB | NumPy | Number theoryFlexible hours | Freelance opportunities | Project based workSenior-level FreelanceNew York, New York, United States … R13h ago
-
Statistics & Python Expert - Freelance AI Trainer USD 146K-146KCombinatorics | Graph theory | Mathematics | NumPy | Number theoryFreelance opportunity | Part-time project-based workSenior-level FreelanceFlorida, United States - Remote R13h ago
-
Statistics & Python Expert - Freelance AI Trainer USD 146K-146KCMME | MATLAB | NumPy | Pandas | PythonFlexible schedule | Part-time project-based work | Project-based compensationSenior-level FreelanceUnited States - Remote R13h ago
-
Statistics & Python Expert - Freelance AI Trainer USD 146K-146KC# | Combinatorics | Graph theory | MATLAB | NumPyFlexible hours | Part-time opportunities | Project based workSenior-level FreelanceTexas, United States - Remote R13h ago
-
Statistics & Python Expert - Freelance AI Trainer USD 146K-146KC# | MATLAB | NumPy | Pandas | PythonPart-time availability | Project based workSenior-level FreelanceMichigan, United States - Remote R13h ago
-
ML Ops Engineer USD 174K-226KAWS | Cloud infrastructure | Cost Optimization | Data Ingestion | GCPHybrid work schedule | In-office at least 3 days per weekMid-level Full TimeSan Francisco HQ Office R14h ago
-
Machine Learning Engineer - 1 USD 130K-228KCNN | Cross-validation | Data Pipelines | Deep learning | Document processingEquity options | Flexible-hybrid work | Medical, dental & vision coverage | Professional development budget | Team offsitesNone Full TimeHybrid - San Mateo, California R14h ago
-
Lead AI Engineer - AI & Credit Analytics USD 156K-234KAWS | CI/CD | Data Governance | Generative AI | LLMOpsFlexible time off | Flexible work environment | Hybrid work option | Matching 401k | Medical/Dental/Vision insuranceSenior-level Full TimeCosta Mesa, CA, United States R15h ago
-
AI Software Engineer - Greenwood Village, CO Office USD 80K-120KAI Agents | API | Automation | C# | Computer VisionCollaborative environment | Comprehensive benefits package | Employee ownership | Flexible workplace | Innovative cultureEntry-level Full TimeGreenwood Village, Colorado, United States R16h ago
-
Data Engineer USD 130K-145KApache Spark | CI/CD | Cloud platform | Containerization | Data GovernancePublic trust clearance support | Remote workSenior-level Full TimeWork from home, VA, United States R20h ago
-
Sr. Embedded Software Developer USD 130K-160KARM | Assembly | Bootloader development | C# | C++401k match | Employee assistance program | Medical, dental & vision coverage | Paid Holidays | Paid time offSenior-level Full TimeRemote (United States) R20h ago
-
Embedded Software Developer USD 110K-140KARM | Bootloaders | C# | C++ | CI/CD401k matching | Career growth | Dental insurance | Employee assistance program | Medical insuranceMid-level Full TimeRemote (United States) R20h ago
-
Sr. Analytics Engineer, Business Intelligence USD 115K-145KDBT | Data Architecture | Data Modeling | Data Quality | Data Warehouse401k | Dental insurance | Discounts and perks | Medical insurance | Paid leaveSenior-level Full TimeNew York, NEW YORK, United States R21h ago
-
DNS | FC | Fibre Channel | Isilon | LinuxRemote work | Unlimited growthSenior-level Full TimeUnited States, United States R22h ago
-
API Integration | Asynchronous processing | Chatbots | Deep learning | Distributed Systems100% remote | Flexible scheduleMid-level Full TimeAnnapolis, Maryland, United States R1d ago
-
AI/ML Engineer, Senior - WFH1650 USD 128K-201KCPU Inference | Class imbalance | Data Analysis | Data Preprocessing | Data QualityWork from homeSenior-level Full TimeReston, VA - Remote R1d ago
-
Mid-level Full TimeRemote, United States R1d ago
-
Apache Airflow | AtScale | BigQuery | CI/CD | Cloud DataSenior-level Full TimeGEORGIA - VIRTUAL - GA01, United … R1d ago
-
Airflow | BigQuery | CI/CD | Data Modeling | Data WarehousingRemoteMid-level Full TimeGEORGIA - VIRTUAL - GA01, United … R1d ago
-
Principal Machine Learning Revenue Architect USD 163K-382KBig Data | Business Process | Data Mining | Data Science | Data VisualizationSenior-level Full TimeVirtual, United States R1d ago
-
Senior Analytics Engineer USD 87K-161KData Lakehouse | Data Modeling | Data mesh | Data pipeline | Databricks401k | Health insurance | Hybrid work options | Paid Holidays | Paid time offSenior-level Full TimeRemote-MO, United States R1d ago
-
Senior Engineer - LLMOps & MLOps USD 120K-189KAI Search | AWS | AWS Step Functions | Airflow | Amazon BedrockCareer growth | Great Place to Work culture | Work-life balanceSenior-level Full TimeTelecommuter TN, United States R1d ago
-
Machine Learning Engineer 5 - Globalization USD 466K-750KBatching | Data Pipelines | Distributed Training | GPU Optimization | Inference Optimization401k matching | Disability coverage | Flexible spending account | Flexible time off | Health insuranceSenior-level Full TimeUSA - Remote, United States R1d ago
-
Sr/Staff AI Engineer (Remote - US) USD 165K-300KAWS | AWS S3 | Agentic architecture | Azure | Databricks401k | Annual bonus | Counseling services | Dental insurance | Disability insuranceSenior-level Full TimeREMOTE, US, US, 76131 R1d ago
-
Sr/Staff Data Engineer (Remote - US) USD 165K-300KAWS Kinesis | Apache Flink | Apache Iceberg | Apache Kafka | Apache Spark401k | Annual bonus | Counseling services | Dental insurance | Disability insuranceSenior-level Full TimeREMOTE, US, US, 76131 R1d ago