Urgent! Machine Learning Engineer (Diffusion / Vision) Position in Petaling Jaya - Bjak
At Bjak, we’re building AI systems that reach millions across Southeast Asia.
Our Machine Learning team is pushing the frontier of image generation, personalization, and safe deployment of diffusion models.
From powering creative assets to enabling user‑facing visual tools, we make diffusion models fast, scalable, and aligned with our product vision.
This role is based in Malaysia with a hybrid work arrangement – combining flexible remote work with in‑office collaboration at our HQ.
You’ll work closely with regional teams across product, engineering, and design to bring cutting‑edge generative vision features to market.
We are scaling generative vision capabilities that reshape how people experience financial and consumer products.
As a Machine Learning Engineer (Diffusion / Vision), you’ll fine‑tune diffusion models, design evaluation frameworks, and deploy AI features into production.
Your work ensures our image systems are creative, safe, and optimized for real‑world impact.
Fine‑tune & Adapt – Train and customize diffusion models (SDXL, Flux, Stable Diffusion variants) using LoRA, DreamBooth, and other parameter‑efficient methods.
Curate Datasets – Build, clean, and annotate large‑scale image datasets with captioning, tagging, and NSFW filtering for safe and aligned generation.
Evaluate & Align – Develop pipelines to measure fidelity, diversity, style adherence, and safety across generated outputs.
Optimize Performance – Apply GPU memory optimization, latent diffusion tricks, and distributed training for efficient scaling.
Deploy & Monitor – Ship diffusion‑powered features into production with monitoring for drift, latency, and quality.
Collaborate & Deliver – Work with product and design to integrate generative vision capabilities into user experiences.
Treat every challenge as your own, regardless of title.
Enjoy switching between researcher, engineer, and product thinker.
Prototype fast – test, iterate, and improve without waiting for perfect conditions.
Stay calm and effective in startup chaos.
Bias for speed – believe something valuable now beats perfect later.
Learn fast from feedback and failure.
Care deeply about both user experience and technical excellence.
Show up with humility, hunger, and hustle.
Strong experience with diffusion models and generative vision (Stable Diffusion, SDXL, Flux, etc.).
Hands‑on skills with DreamBooth, LoRA/QLoRA, and fine‑tuning methods.
Proficiency with PyTorch (preferred).
Experience in dataset preparation (captioning, tagging, filtering, augmentation).
Knowledge of GPU optimization, latent diffusion, and efficient training techniques.
Strong foundations in software engineering, algorithms, and clean code practices.
Prior work in style transfer, personalization, or multimodal generation.
Experience with safety alignment in generative vision (NSFW detection, adversarial testing).
Open‑source contributions in diffusion or generative vision projects.
Competitive salary and performance‑based bonuses
Flexible work arrangement
High‑impact role with visibility across product, design, and engineering
Flat structure – your work is seen, your ideas are heard
Accelerated learning and growth in a mission‑driven company
Cross‑border exposure to regional product development
We’re a lean, high‑performance team that moves fast and holds a high bar.
Titles don’t matter here – output, integrity, and ownership do.
Everyone is expected to roll up their sleeves, contribute meaningfully, and take initiative.
We reward people who think like owners.
We value speed, clarity, and relentless execution.
If you’re hungry to grow and care deeply about excellence, this is your kind of environment.
BJAK is Southeast Asia’s #1 insurance aggregator with 8M+ users, fully owned by its employees.
Headquartered in Malaysia and operating in Thailand, Taiwan, and Japan, we help millions of users access transparent and affordable financial protection through Bjak.com.
We simplify complex financial products through cutting‑edge technologies, including APIs, automation, and AI, to build the next generation of intelligent financial systems.
If you’re excited to build real‑world AI systems and grow fast in a high‑impact environment, we’d love to hear from you.
✨ Smart • Intelligent • Private • Secure
Practice for Any Interview Q&A (AI Enabled)
Predict interview Q&A (AI Supported)
Mock interview trainer (AI Supported)
Ace behavioral interviews (AI Powered)
Record interview questions (Confidential)
Master your interviews
Track your answers (Confidential)
Schedule your applications (Confidential)
Create perfect cover letters (AI Supported)
Analyze your resume (NLP Supported)
ATS compatibility check (AI Supported)
Optimize your applications (AI Supported)
O*NET Supported
O*NET Supported
O*NET Supported
O*NET Supported
O*NET Supported
European Union Recommended
Institution Recommended
Institution Recommended
Researcher Recommended
IT Savvy Recommended
Trades Recommended
O*NET Supported
Artist Recommended
Researchers Recommended
Create your account
Access your account
Create your professional profile
Preview your profile
Your saved opportunities
Reviews you've given
Companies you follow
Discover employers
O*NET Supported
Common questions answered
Help for job seekers
How matching works
Customized job suggestions
Fast application process
Manage alert settings
Understanding alerts
How we match resumes
Professional branding guide
Increase your visibility
Get verified status
Learn about our AI
How ATS ranks you
AI-powered matching
Join thousands of professionals who've advanced their careers with our platform
Unlock Your Machine Learning Potential: Insight & Career Growth Guide
Real-time Machine Learning Jobs Trends in Petaling Jaya, Malaysia (Graphical Representation)
Explore profound insights with Expertini's real-time, in-depth analysis, showcased through the graph below. This graph displays the job market trends for Machine Learning in Petaling Jaya, Malaysia using a bar chart to represent the number of jobs available and a trend line to illustrate the trend over time. Specifically, the graph shows 562 jobs in Malaysia and 55 jobs in Petaling Jaya. This comprehensive analysis highlights market share and opportunities for professionals in Machine Learning roles. These dynamic trends provide a better understanding of the job market landscape in these regions.
Great news! Bjak is currently hiring and seeking a Machine Learning Engineer (Diffusion / Vision) to join their team. Feel free to download the job details.
Wait no longer! Are you also interested in exploring similar jobs? Search now: Machine Learning Engineer (Diffusion / Vision) Jobs Petaling Jaya.
An organization's rules and standards set how people should be treated in the office and how different situations should be handled. The work culture at Bjak adheres to the cultural norms as outlined by Expertini.
The fundamental ethical values are:The average salary range for a Machine Learning Engineer (Diffusion / Vision) Jobs Malaysia varies, but the pay scale is rated "Standard" in Petaling Jaya. Salary levels may vary depending on your industry, experience, and skills. It's essential to research and negotiate effectively. We advise reading the full job specification before proceeding with the application to understand the salary package.
Key qualifications for Machine Learning Engineer (Diffusion / Vision) typically include Engineering and a list of qualifications and expertise as mentioned in the job specification. Be sure to check the specific job listing for detailed requirements and qualifications.
To improve your chances of getting hired for Machine Learning Engineer (Diffusion / Vision), consider enhancing your skills. Check your CV/Résumé Score with our free Resume Scoring Tool. We have an in-built Resume Scoring tool that gives you the matching score for each job based on your CV/Résumé once it is uploaded. This can help you align your CV/Résumé according to the job requirements and enhance your skills if needed.
Here are some tips to help you prepare for and ace your job interview:
Before the Interview:To prepare for your Machine Learning Engineer (Diffusion / Vision) interview at Bjak, research the company, understand the job requirements, and practice common interview questions.
Highlight your leadership skills, achievements, and strategic thinking abilities. Be prepared to discuss your experience with HR, including your approach to meeting targets as a team player. Additionally, review the Bjak's products or services and be prepared to discuss how you can contribute to their success.
By following these tips, you can increase your chances of making a positive impression and landing the job!
Setting up job alerts for Machine Learning Engineer (Diffusion / Vision) is easy with Malaysia Jobs Expertini. Simply visit our job alerts page here, enter your preferred job title and location, and choose how often you want to receive notifications. You'll get the latest job openings sent directly to your email for FREE!
