companyBjak logo

Technical Lead in Machine Learning

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

QualificationsProficiency in Python and familiarity with ML frameworks such as PyTorch or JAX. Hands-on experience with GPU-based training and inference systems. Excellent coding skills with an emphasis on system correctness and reliability. Ability to work autonomously and collaboratively within a team. Strong problem-solving skills and a results-oriented mindset.

About the job

About the Role

Join bjakcareer as we innovate a cutting-edge AI system that comprehends context throughout conversations, orchestrates actions, and advances tasks over time.

In this pivotal role, you will transform research insights into robust, production-ready machine learning systems. You will oversee the execution layer of our AI's intelligence, encompassing training pipelines, inference systems, evaluation tools, and deployment strategies.

What You'll Do

  • Develop and manage comprehensive ML pipelines, integrating data collection, training, evaluation, inference, and deployment.

  • Optimize and tailor models utilizing advanced techniques including LoRA, QLoRA, SFT, DPO, and model distillation.

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

  • Create and sustain data systems that yield high-quality synthetic and real-world training data.

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

  • Lead production deployment efforts, focusing on GPU optimization, memory efficiency, latency reduction, and scaling policies.

  • Work closely with application engineers to seamlessly integrate ML systems into backend, mobile, and desktop products.

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

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

Tech Stack

  • Proficient in Python

  • Experience with PyTorch or JAX

  • Familiarity with GPU-based training and inference systems

Ideal Experience

  • Demonstrated experience building or deploying real ML systems that users engage with, beyond demos.

  • Comfortable managing large models and understanding their limitations.

  • Strong ability to write production-grade code with a focus on system reliability.

  • Self-motivated, pragmatic, and accountable for results.

  • Excellent communication skills and ability to collaborate effectively in small, high-trust teams.

How We Work

Our organization fosters a culture of innovation and collaboration, where every team member is encouraged to contribute their unique perspectives and expertise.

About Bjak

bjakcareer is at the forefront of AI innovation, dedicated to creating intelligent systems that enhance user experiences. Our mission is to build proactive AI technologies that empower users by understanding context and facilitating seamless interactions.

Similar jobs

Tailoring 0 resumes

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