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 background in deep learning and transformer architectures
Hands-on experience training or fine-tuning large models (LLMs or vision models)
Proficiency with Py Torch, JAX, or Tensor Flow
Experience with distributed training frameworks (Deep Speed, FSDP, Megatron, Ze RO, 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
Nice to Have
Experience with LLM inference frameworks (v LLM, Tensor RT-LLM, Faster Transformer)
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, Deep Mind, FAIR, Anthropic, Open AI)
What You'll Get
Extreme ownership and autonomy from day one - you define and build key model systems.
Founding-level influence over technical direction, model architecture, and product strategy.
Remote-first flexibility
High-impact scope—your work becomes core infrastructure of a global consumer AI product.
Competitive compensation and performance-based bonuses
Backing of a profitable US$2 B 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
A small, senior, high-performance team where you collaborate directly with founders and influence every major decision.
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 group backed by BJAK, focused on building a new consumer AI product with global impact.
We're assembling a small, elite team of ML and engineering builders who want to work on meaningful, high-impact problems.
#J-18808-Ljbffr