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.

