AI Architecture Research Engineer — Build Next-Gen Models

Featherless AI; Poland; Remote; PLN 80,000 - 120,000; Full time

Benefits offered by this job

Competitive compensation

Meaningful equity

Direct influence on technical direction

Job summary

A Series-A technology company in Poland is seeking a Machine Learning Engineer focused on AI architecture research. The ideal candidate will design, prototype, and validate new neural network architectures while collaborating with engineers to ensure deployable and efficient models. Strong fundamentals in machine learning, hands-on experience in model architecture implementation, and proficiency in PyTorch or JAX are essential. This role offers competitive compensation and meaningful equity in a small, high-caliber team.

Qualifications

• Strong background in machine learning fundamentals and deep learning.
• Hands‑on experience implementing model architectures from scratch.
• Comfortable working in PyTorch or JAX.

Responsibilities

• Research and develop new neural network architectures.
• Design and run architecture-level experiments.
• Collaborate with inference and systems engineers.

Skills

Machine learning fundamentals

Deep learning

Model architectures implementation

Attention mechanisms

Experience in PyTorch or JAX

Training dynamics

Tools

PyTorch

JAX

Job description

A Series-A technology company in Poland is seeking a Machine Learning Engineer focused on AI architecture research. The ideal candidate will design, prototype, and validate new neural network architectures while collaborating with engineers to ensure deployable and efficient models. Strong fundamentals in machine learning, hands-on experience in model architecture implementation, and proficiency in PyTorch or JAX are essential. This role offers competitive compensation and meaningful equity in a small, high-caliber team.