BJAK · 1 month ago
Founding AI Engineer
United States
Full-time
Remote
Senior LevelBJAK is a self-funded AI group focused on building a new global consumer AI application. The Founding AI Engineer will shape the core technical direction of the AI group, including model selection, training strategy, and infrastructure while creating impactful product experiences.
Information TechnologyInsurTechPrice Comparison
Responsibilities
Build end-to-end training pipelines: data → training → eval → inference
Design new model architectures or adapt open-source frontier models
Fine-tune models using state-of-the-art methods (LoRA/QLoRA, SFT, DPO, distillation)
Architect scalable inference systems using vLLM / TensorRT-LLM / DeepSpeed
Build data systems for high-quality synthetic and real-world training data
Develop alignment, safety, and guardrail strategies
Design evaluation frameworks across performance, robustness, safety, and bias
Own deployment: GPU optimization, latency reduction, scaling policies
Shape early product direction, experiment with new use cases, and build AI-powered experiences from zero
Explore frontier techniques: retrieval-augmented training, mixture-of-experts, distillation, multi-agent orchestration, multimodal models
Qualification
Required
Strong background in deep learning and transformer architectures
Hands-on experience training or fine-tuning large models (LLMs or vision models)
Proficiency with PyTorch, JAX, or TensorFlow
Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)
Strong software engineering skills — writing robust, production-grade systems
Experience with GPU optimization: memory efficiency, quantization, mixed precision
Comfortable owning ambiguous, zero-to-one technical problems end-to-end
Preferred
Experience with LLM inference frameworks (vLLM, TensorRT-LLM, FasterTransformer)
Contributions to open-source ML libraries
Background in scientific computing, compilers, or GPU kernels
Experience with RLHF pipelines (PPO, DPO, ORPO)
Experience training or deploying multimodal or diffusion models
Experience in large-scale data processing (Apache Arrow, Spark, Ray)
Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI)
Benefits
Insurance coverage
Flexible time off
Global travel insurance
Company
BJAK
Bjak is a technology company that develops an insurance comparison platform.
Founded in 2019
201-500 employees
Funding
Current Stage
Growth StageLeadership Team
Kuan Yue Ronald Lim
Executive Director
Recent News
Fintech News Malaysia
2025-12-16
2025-12-16
2025-11-29
Company data provided by crunchbase