Founding ML Architect for Global Consumer AI

Bjak; Remote; PLN 240,000 - 320,000; Full time

Job summary

An innovative AI group in Poland is seeking a founding technical contributor to drive the development of an impactful consumer AI application. In this role, you will have significant freedom to design core systems and make key decisions about model architectures and data strategies. Ideal candidates will possess a strong background in deep learning and have experience in building scalable systems. This position offers remote-first flexibility, competitive compensation, and the opportunity to shape a product with global significance.

Benefits

Competitive compensation

Flexible time off

Insurance coverage

Global travel insurance

Founding-level influence

Qualifications

• Strong background in deep learning and transformer architectures.
• Hands-on experience training or fine-tuning large models.
• Proficiency with frameworks like PyTorch, JAX, or TensorFlow.
• Experience with distributed training frameworks.
• Strong software engineering skills for production systems.
• Experience with GPU optimization techniques.
• Ability to handle ambiguous technical problems.

Responsibilities

• Build training pipelines for data, training, evaluation, and inference.
• Design and implement new model architectures or adapt existing ones.
• Fine-tune models using state-of-the-art methods.
• Architect scalable inference systems.
• Develop systems for training data collection and processing.
• Create safety and evaluation strategies for models.
• Manage deployment processes, optimizing for performance.
• Shape product direction and experiment with new use cases.

Skills

Deep learning

Transformer architectures

Fine-tuning large models (LLMs or vision models)

Proficiency with PyTorch

Experience with distributed training frameworks

Software engineering skills

GPU optimization

Problem ownership

Tools

JAX

TensorFlow

DeepSpeed

FSDP

Ray

Job description

An innovative AI group in Poland is seeking a founding technical contributor to drive the development of an impactful consumer AI application. In this role, you will have significant freedom to design core systems and make key decisions about model architectures and data strategies. Ideal candidates will possess a strong background in deep learning and have experience in building scalable systems. This position offers remote-first flexibility, competitive compensation, and the opportunity to shape a product with global significance.