Generative AI - ML System Engineering
Tasks
- Build and scale high throughput 3D data pipelines
- Build data loading framework and libraries
- Build end to end machine learning framework for 3D
- Debug and monitor hardware platform performance
- Design training pipeline for pretraining and finetuning
- Develop inference pipelines for diffusion models
- Implement custom operators with CUDA
- Implement custom operators with Triton
- Optimize distributed model training across GPUs
- Optimize models using compilation
- Optimize models using fusion
- Optimize models using quantization
Perks/Benefits
Skills/Tech-stack
C++ | CUDA | Compilation | Data pipeline | Diffusion Models | Distributed Training | Fusion | GPU Performance | GPU Performance Optimization | JAX | Machine Learning | Model Parallelism | Performance optimization | PyTorch | Python | Quantization | Tensor programming | Triton
Education
N/A
Related jobs
-
Deep Learning Compiler CI/Infrastructure Engineer CNY 160K-240KAI Agents | Agent workflows | Artifact management | Automated triage | AutomationGenerous benefits packageSenior-level Full TimeChina, Shanghai R1d ago
-
Applied AI Engineer CNY 300K-399KA/B | A/B Testing | API Integration | Analytics | AnthropicCareer growth | Fully remote | Global team collaboration | High ownership culture | Learning and development budgetMid-level Full TimeChina R1d ago
-
Lead AI Engineer (AI Systems & Automation) CNY 360K-600KAlerting | Anthropic | Distributed Systems | Docker | EmbeddingsFully remote | High ownership culture | Learning and development budgetSenior-level Full TimeChina R1d ago
-
Mid-level Full Time深圳 R5d ago
-
Mid-level Full Time北京 R5d ago
-
Mid-level Full Time北京 R5d ago
-
MiMo-大模型训练框架开发工程师 CNY 240K-480KC++ | CI/CD | DeepSpeed | Distributed Training | GPU Memory OptimizationEntry-level Full Time北京 R6d ago
-
Entry-level Full Time北京 R6d ago
-
Mid-level Full Time北京 R6d ago
-
具身智能算法工程师-模型 CNY 500K-500KDeep learning | Distributed Training | IQL | Inference Optimization | Isaac LabMid-level Full Time北京 R6d ago
-
AI Engineer USD 100K-200KAPIs | Agentic Frameworks | Artificial Intelligence | Backtesting | Data integrationCollaborative work environment | Equity options | Innovation-driven cultureMid-level Full TimeShanghai, Shanghai, China R7d ago
-
Senior Software Engineer (RAG Backend Developer) CNY 120K-180KA/B | A/B Testing | ABAC | Audit Logging | B testingSenior-level Full TimeGuangzhou, Guangdong, China R8d ago
-
Ai 院--多模态团队--多模态理解算法研究员-强化学习方向 CNY 240K-480KDPO | Data Preprocessing | Data cleaning | DeepSpeed | Distributed TrainingSenior-level Full Time北京 R28d ago
-
Lead Technical Support Engineer - AI / ML CNY 144K-240KAPI Integration | Agent Frameworks | C plus plus | Cause analysis | Cloud ComputingHybrid work model | Travel for customer workshops | Work from homeSenior-level Full TimeBeijing, China R29d ago
-
Agent Development | Agile | Artificial Intelligence | Data Privacy | Data SecurityMid-level Full TimeXi'an, Shaanxi, China R29d ago
-
AI基础设施研发工程师(Sandbox / 容器化)-MiMo CNY 180K-420KAppArmor | Argo Workflows | CI/CD | CPU resource scheduling | CgroupMid-level Full Time北京 R30d ago
-
Entry-level Full Time北京、上海 R1mo ago
-
AI platforms | API Development | Artificial Intelligence | Cloud AI | Cloud AI PlatformsMid-level Full TimeRemote, China R1mo ago
-
AI ML Engineer CNY 280K-360KAWS | Azure | C++ | Cloud Computing | Computer VisionPerformance bonuses | Professional development opportunities | Remote workMid-level Full TimeShenzhen, Guangdong Province, China R1mo ago
-
AI工程师-Agent Memory & RAG 方向(成都) CNY 240K-480KBERT | Chroma | Cross-Encoder | Embedding Models | FaissSenior-level Full Time成都 R1mo ago
-
AI工程师-Agent Memory & RAG 方向(武汉) CNY 240K-480KAlgorithms | BERT | Chroma | Cross-Encoder | Data StructuresSenior-level Full Time武汉 R1mo ago
-
AI工程师-Agent Memory & RAG 方向(北京) CNY 240K-480KBERT | Chroma | Cross-Encoder | Embedding Models | FaissSenior-level Full Time北京 R1mo ago
-
AWS | Agent Orchestration | Agent systems | Azure | DockerMid-level Full TimeShenzhen, Guangdong, China R1mo ago
-
AVP, AI Solution Lead CNY 360K-600KCloud Computing | DataOps | DevOps | Flutter | Generative AIContinuous professional development | Flexible workingSenior-level Full TimeGuangzhou, Guangdong, China R1mo ago
-
AWQ | AWS | Accelerate | Azure | BatchingMid-level Full TimeShenzhen, Guangdong, China R1mo ago