companyBjak logo

Senior Staff Machine Learning Engineer

BjakChina
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Mid to Senior

Qualifications

QualificationsProven experience in developing and deploying ML systems. Strong understanding of machine learning methodologies and practices. Proficient in Python and familiar with frameworks like PyTorch and JAX. Ability to work effectively in a collaborative team environment. Demonstrated ability to communicate complex technical concepts clearly.

About the job

About the Role

A1 is at the forefront of developing an advanced AI system designed to comprehend context across conversations, formulate strategic actions, and maintain continuity over time.

Your role will involve transforming innovative research into operational, production-grade machine learning systems. You will lead the execution layer of A1’s intelligence, overseeing training pipelines, inference systems, evaluation tools, and deployment processes.

What You'll Be Responsible For

  • Develop and manage comprehensive ML pipelines that encompass data management, training, evaluation, inference, and deployment.

  • Optimize and refine models utilizing cutting-edge methodologies such as LoRA, QLoRA, SFT, DPO, and model distillation.

  • Design and maintain scalable inference systems, ensuring a balance between latency, cost-effectiveness, and reliability.

  • Create and manage data systems to ensure access to high-quality synthetic and real-world training data.

  • Establish evaluation pipelines that assess performance, robustness, safety, and bias, collaborating closely with research leadership.

  • Oversee production deployment, focusing on GPU optimization, memory efficiency, latency reduction, and scaling strategies.

  • Engage closely with application engineering teams to seamlessly integrate ML systems into backend, mobile, and desktop applications.

  • Make informed trade-offs and rapidly implement improvements based on real-world usage.

  • Operate under real production constraints including latency, cost, reliability, and safety considerations.

Technical Expertise

  • Proficient in Python

  • Experience with PyTorch and JAX

  • Familiarity with GPU-based training and inference systems

Preferred Qualifications

  • You have successfully built or deployed machine learning systems that deliver real value to users.

  • You possess a strong understanding of large models and their potential failure modes.

  • Your coding skills are robust, with a focus on production-grade quality and system integrity.

  • Self-motivated and pragmatic, you take full ownership of your contributions and outcomes.

  • You communicate effectively and excel in collaborative, high-trust team environments.

Our Working Environment

Our organization is structured to foster innovation and collaboration among talented individuals, committed to pushing the boundaries of AI technology.

About Bjak

bjakcareer is a leading technology firm dedicated to leveraging AI innovations to create transformative solutions. We are committed to building systems that enhance user experiences and drive efficiency across various sectors.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.