Software Engineer, AI and DL Kernel Libraries
Tasks
- Analyze performance profile and optimize workloads
- Build deep learning library abstractions
- Build just in time compilation and code generation
- Collaborate with compilers and GPU architecture teams
- Contribute to open source inference ecosystems
- Design implement optimize GPU kernels
- Develop production AI inference software
- Integrate and improve LLM inference runtimes
Perks/Benefits
- N/A
Skills/Tech-stack
API Design | Apache TVM | C# | C++ | CUDA | CUDA C++ | CUDNN | Code generation | Code optimization | GPU Performance | GPU performance modeling | JAX | Just-in-Time | Just-in-time compilation | Linear Algebra | MLIR | ONNX | Performance Modeling | Performance Profiling | PyTorch | Python | System Architecture | TensorFlow | TensorRT | Triton
Education
Bachelor of Engineering | Bachelor of Science | Master of Science
Related jobs
-
Entry-level Internship上海9h ago
-
多模态大模型数据工程师 CNY 25K-37KCorpus | Data Augmentation | Data Quality | Data Validation | Data pipelineInternship experienceEntry-level Internship北京9h ago
-
Mid-level Full Time上海9h ago
-
AVP, AI Solution Lead CNY 144K-240KArtificial Intelligence | Cloud Native | DataOps | DevOps | FlutterContinuous professional development | Flexible workingSenior-level Full TimeGuangzhou, Guangdong, China R18h ago
-
Active Directory | Amazon Web Services | Ansible | Apache Airflow | Apache KafkaMid-level Full TimeShanghai, SH, CN1d ago
-
AVP, AI Solution Lead CNY 144K-240KCloud Native | Cloud deployment | DataOps | DevOps | FlutterContinuous professional development | Flexible working | Inclusive work environmentSenior-level Full TimeGuangzhou, GD, CN, 5106201d ago
-
Entry-level Full Time深圳1d ago
-
None Full Time上海、深圳、武汉、北京1d ago
-
提前批-Ai数据产品经理 CNY 144K-240KAPI Integration | Agentic Workflows | Knowledge graph | LLM | Ontology ModelingSenior-level Full Time上海1d ago
-
None Full Time北京、上海1d ago
-
Mid-level Full Time上海、圣何塞1d ago
-
None Full Time上海、北京、深圳1d ago
-
Senior-level Full Time上海、北京、深圳1d ago
-
Entry-level Full Time上海1d ago
-
机器人交互数据与评测工程师 CNY 36K-42KBenchmarking | Data Governance | Data Mining | Data Quality | Data VersioningNone Full Time上海1d ago
-
None Full Time上海1d ago
-
Deep Learning Performance Architect CNY 144K-240KComputer Architecture | Deep learning | Hardware performance | Hardware performance analysis | Inference OptimizationSenior-level Full TimeChina, Shanghai1d ago
-
Senior-level Full Time上海1d ago
-
Entry-level Full Time广州、上海1d ago
-
Entry-level Internship苏州1d ago
-
AI Foundation Model方向实习生 CNY 25K-37KC++ | Caffe | Computer Vision | Data Structures | Data Structures and AlgorithmsEntry-level Internship北京、上海1d ago
-
Ai产品经理 CNY 144K-240KArtificial Intelligence | Data analytics | Data cleaning | Data collection | Data synchronizationEntry-level Full Time深圳1d ago
-
Mid-level Full Time深圳 R1d ago
-
【27届校招提前批】算法工程师 CNY 180K-360KAgent collaboration | Asynchronous programming | Caching | Concurrency | Context engineeringNone Full Time上海1d ago
-
Entry-level Full Time北京1d ago