Founding Machine Learning Engineer

BJAK


Date: 3 weeks ago
City: Petaling Jaya
Contract type: Full time
Transform Language Models Into Real-world, High-impact Product Experiences.

A1 is a self-funded AI division backed by BJAK, operating in full stealth. We’re building a new global consumer AI application focused on an important but underexplored use case — something practical, meaningful, and far beyond the typical chatbot or productivity agent.

As a Machine Learning Engineer, you will work at the core of our product: fine-tuning frontier-level models, shaping data pipelines, evaluating behavior, and bringing intelligent features into real production environments. You’ll operate with high autonomy within a small, deeply technical team.

Why This Role Matters

  • You will turn state-of-the-art models into real product capabilities used by a global audience.
  • You’ll define how our models behave — accuracy, safety, tone, and user experience.
  • You’ll help build a new consumer AI product from zero, directly influencing its technical and product direction.
  • Your work will ensure our models are not only powerful, but safe, reliable, and impactful at scale.

What You’ll Do

  • Fine-tune and adapt models as needed while also designing new model architectures and training pipelines from scratch
  • Design and curate datasets, craft prompts, and experiment with alignment and safety strategies for high-quality outputs.
  • Build evaluation pipelines for model performance, safety, and alignment, and innovate on new metrics for real-world impact.
  • Ship models to production, optimise for latency and scale, and monitor drift or degradation
  • Collaborate with product, engineering, and design to launch polished, user-facing AI features
  • Explore new modeling approaches, optimisation methods, and product use cases

What It’s Like To Work Here

  • You take ownership - you solve problems end-to-end rather than wait for perfect instructions
  • You learn through action - prototype → test → iterate → ship
  • You’re calm in ambiguity - zero-to-one building energises you
  • You bias toward speed with discipline - V1 now > perfect later
  • You see failures and feedback as essential to growth
  • You work with humility, curiosity, and a founder’s mindset
  • You lift the bar for yourself and your teammates every day

Requirements

  • Strong experience with transformers, deep learning, and fine-tuning methods (LoRA/QLoRA, SFT, distillation)
  • Proficiency in PyTorch (preferred) or TensorFlow
  • Experience in prompt engineering and dataset creation for alignment, tone, trust, and safety
  • Familiarity with evaluation methods: perplexity, toxicity, relevance, robustness
  • Solid software engineering fundamentals — algorithms, data structures, clean code
  • Ability to operate in a fast, lean, and high-ownership environment

Nice to Have

  • Experience with text generation, moderation, ranking, or personalisation
  • Exposure to RLHF or reinforcement learning for LLMs
  • Contributions to open-source ML projects
  • Background in consumer-facing AI products or high-growth startups

What You’ll Get

  • High ownership and autonomy from day one
  • Direct involvement in technical direction and product decisions
  • Remote-first flexibility
  • A high-impact role with visibility across ML, engineering, and product
  • Competitive compensation and performance-based bonuses
  • Backing of a profitable US$2B group, with the speed of a startup
  • Insurance coverage, flexible time off, and global travel insurance
  • Opportunity to shape a new global AI product from zero

Our Team & Culture

We operate as a dense, senior, high-performance team. We value clarity, speed, craftsmanship, and relentless ownership. We behave like founders — we build, ship, iterate, and hold ourselves to a high technical bar.

If you value excellence, enjoy building real systems, and want to be part of a small team creating something globally impactful, you’ll thrive here.

About A1

A1 is a self-funded, independent AI division under BJAK, focused on building a new consumer AI product with global relevance. We’re in full stealth and assembling a small, elite team of ML and engineering builders who want to work on meaningful, high-impact problems.

How to apply

To apply for this job you need to authorize on our website. If you don't have an account yet, please register.

Post a resume

Similar jobs

Service Delivery Field Support Engineer (L1)

NTT DATA, Inc., Petaling Jaya
13 hours ago
Make an impact with NTT DATAJoin a company that is pushing the boundaries of what is possible. We are renowned for our technical excellence and leading innovations, and for making a difference to our clients and society. Our workplace embraces diversity and inclusion – it’s a place where you can grow, belong and thrive.Your day at NTT DATAThe Cross Technology...

SAP ERP Senior Consultant – Order to Cash

Evonik, Petaling Jaya
2 days ago
What We OfferExplore a world of opportunities with us. Look ahead with us and help shape innovative solutions to make our world more sustainable and life healthier, more vibrant and more comfortable. At Evonik, you have the chance to explore, thrive, and grow alongside 33,000 colleagues. Among attractive career paths and high-quality development programs, we not only offer performance-based remuneration...

PROTÉGÉ Trainee

Abbott, Petaling Jaya
4 days ago
About AbbottAbbott is a global healthcare leader, creating breakthrough science to improve people’s health. We’re always looking towards the future, anticipating changes in medical science and technology.Working at Abbott At Abbott, you can do work that matters, grow, and learn, care for yourself and family, be your true self and live a full life. You will have access to :A...