Machine Learning Engineer, AML - Engine (Multiple Positions)

关于这个机会

About ByteDance Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of more than a dozen products, including TikTok, Lemon8, CapCut and Pico as well as platforms specific to the China market, including Toutiao, Douyin, and Xigua, ByteDance has made it easier and more fun for people to connect with, consume, and create content. Why Join Us Inspiring creativity is at the core of ByteDance's mission. Our innovative products are built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and enrich life - a mission we work towards every day. As ByteDancers, we strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our Company, and our users. When we create and grow together, the possibilities are limitless. Join us. About the Team Our team plays a crucial role in ensuring the company’s success. We seek people who are willing to learn and put in the effort to solve problems. Our challenges are not your regular day-to-day problems - you’ll be part of a team that’s developing new solutions to new challenges. It’s working fast, at scale, and we’re making a difference. We are looking for talents to join us on this exciting journey! Responsibilities Apply machine learning and robotics research knowledge including LLM and reinforcement learning to analyze and optimize deep learning model architectures. Design and implement high-performance CUDA kernels to accelerate model inference and computation-heavy operators. Analyze computation graphs to identify optimization opportunities across different model structures. Leverage parallel computing techniques such as CUDA and OpenMP to improve throughput, latency, and resource utilization in ML workloads. Migrate between TensorFlow and PyTorch model serving frameworks, ensuring correctness and performance consistency. Develop algorithm-level optimizations for both in-graph and out-of-graph computations, balancing AUC and execution efficiency.

任职要求

Qualifications Must have a Master's degree or foreign equivalent degree in Computer Science, Engineering (any), Information Technology, Data Science, Robotics, Mathematics, or a related quantitative field, and 1 year of related work experience; OR a Bachelor's degree or foreign equivalent degree in Computer Science, Engineering (any), Information Technology, Data Science, Robotics, Mathematics, or a related quantitative field, and 3 years of related work experience. Of the required experience, must have 1 year of experience in each of the following: Coding using Python or C++; Optimize deep learning training and inference pipelines on GPU and CPU architectures; Implement CUDA-based performance optimizations for machine learning workloads; Improve systems-level performance across machine learning frameworks; Developing and deploying large-scale machine learning systems; and Utilizing Big data framework using MySQL, Spark, Hadoop, or Flink. Type: Full time, 40 hours/week Location: San Jose, CA Salary Range: $187741 - $316800 per year To Apply, click the apply button below. Contact lpresumes@bytedance.com if you have difficulty submitting resume through the website. ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At ByteDance, our mission is to inspire creativity and enrich life. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too. ByteDance is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at https://tinyurl.com/RA-request #IND-DNI

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。