BJAK logo
BJAK

Bjak is a technology company focused on making financial services easy, fun and more rewarding for everyone

Principal Machine Learning Engineer

Machine Learning EngineerMachine Learning EngineerFull TimeRemoteLeadTeam 51-200H1B No SponsorCompany SiteLinkedIn

Location

United Kingdom

Posted

31 days ago

Salary

0

Seniority

Lead

Bachelor DegreeEnglishPyTorchRay

Job Description

Principal Machine Learning Engineer

BJAK

• Build and own end-to-end ML pipelines spanning data, training, evaluation, inference, and deployment. • Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation. • Architect and operate scalable inference systems, balancing latency, cost, and reliability. • Design and maintain data systems for high-quality synthetic and real-world training data. • Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership. • Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies. • Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products. • Make pragmatic trade-offs and ship improvements quickly, learning from real usage. • Work under real production constraints: latency, cost, reliability, and safety

Job Requirements

  • Strong background in deep learning and transformer-based architectures.
  • Hands-on experience training, fine-tuning, or deploying large-scale ML models in production.
  • Proficiency with at least one modern ML framework (e.g. PyTorch, JAX), and ability to learn others quickly.
  • Experience with distributed training and inference frameworks (e.g. DeepSpeed, FSDP, Megatron, ZeRO, Ray).
  • Strong software engineering fundamentals – you write robust, maintainable, production-grade systems.
  • Experience with GPU optimization, including memory efficiency, quantization, and mixed precision.
  • Comfort owning ambiguous, zero-to-one ML systems end-to-end.
  • A bias toward shipping, learning fast, and improving systems through iteration.

Benefits

  • Applications are evaluated by our technical team members
  • Interviews will be conducted via virtual meetings and/or onsite
  • We value transparency and efficiency, so expect a prompt decision.

Related Job Pages

More Machine Learning Engineer Jobs

Inspiren logo

Senior Machine Learning Engineer

Inspiren

Inspiren offers the most complete and connected ecosystem in senior living. Founded by Michael Wang, a former Green Beret turned cardiothoracic nurse, Inspiren proves that compassionate care and technology can coexist - bringing peace of mind to residents, families, and staff. Our integrated solutions seamlessly fit into existing workflows, capturing everything happening within a community. Backed by nurse specialists and powerful analytics, we provide the data operators need to make informed clinical and operational decisions - driving efficiency, profitability, and better care outcomes.

Full TimeRemoteTeam 1-10Since 2016H1B No Sponsor

• Design, build, and iterate on VLM-based pipelines that generate high-quality labels and annotations at scale, including prompt engineering, fine-tuning, and evaluation • Determine which signals, frames, metadata, and contextual features should be sent from edge devices to improve VLM accuracy and reduce ambiguity • Collaborate with Embedded Systems and Hardware teams to define device-side preprocessing and data-forwarding strategies that balance bandwidth, latency, and model performance • Collaborate with Data Science to build robust evaluation frameworks to measure label quality, model accuracy, and regression detection • Benchmark and integrate commercial and open-source VLMs, staying current on the fast-moving landscape of vision-language capabilities

New York
$200K - $230K / year
Full TimeRemoteTeam 11-50Since 2018

• Own the lifecycle of our computer vision models, including training, evaluation, deployment, and iteration • Improve model performance in real-world conditions (noise, edge cases, data drift) • Design and improve data pipelines that process thousands of miles of sidewalks and millions of images • Lead architectural decisions for handling hundreds of terabytes of geospatial and visual data, including storage layout, pipeline reliability, and inference performance • Set and evolve best practices around deployment, observability, system reliability, and scalability • Act as a senior individual contributor and mentor, helping raise the technical bar across the engineering team • Work closely with the CEO and operations team to turn business needs into clear technical priorities

United States
$150K - $190K / year
Provectus logo

Senior ML Engineer – GenAI, AWS

Provectus

We help businesses leverage cloud, data, and AI to reimagine the way they operate, compete, and deliver customer value.

Full TimeRemoteTeam 501-1,000Since 2012H1B Sponsor

• Design and implement end-to-end ML solutions from experimentation to production; • Build scalable ML pipelines and infrastructure; • Optimize model performance, efficiency, and reliability; • Write clean, maintainable, production-quality code; • Conduct rigorous experimentation and model evaluation; • Troubleshoot and resolve complex technical challenges. • Mentor junior and mid-level ML engineers; • Conduct code reviews and provide constructive feedback; • Share knowledge through documentation, presentations, and workshops; • Collaborate with cross-functional teams (DevOps, Data Engineering, SAs); • Contribute to internal ML practice development. • Stay current with ML research and emerging technologies; • Propose improvements to existing solutions and processes; • Contribute to the development of reusable ML accelerators; • Participate in technical discussions and architectural decisions.

Colombia
MindTech logo

Machine Learning Engineer, Ssr

MindTech

Your partner in Nearshore IT Outsourcing services. From Latam to the world.

Full TimeRemoteTeam 51-200Since 2021H1B Sponsor

• Diseñar e implementar soluciones basadas en LLMs (GPT, Claude u otros). • Construir agentes y flujos conversacionales, gestionando contexto, prompts y memoria. • Integrar servicios de IA a través de APIs (OpenAI, Anthropic, Azure AI). • Trabajar con datos estructurados y no estructurados para enriquecer las capacidades del sistema. • Iterar sobre modelos y prompts en base a feedback de usuarios y métricas. • Optimizar performance, calidad de respuestas y costo de inferencia. • Asegurar buenas prácticas de seguridad, privacidad y uso responsable de IA. • Colaborar con equipos de producto y negocio para definir casos de uso y soluciones.

Colombia