AI Research Engineer

About Us

At Rossum, our mission is to leverage cutting-edge AI technology and an intuitive UI to eliminate needless paperwork, empowering businesses worldwide to move faster, which we believe ultimately contributes to a better world.

Our revolutionary document processing platform helps hundreds of customers automate data extraction and improve communication efficiency every day. By advancing the way companies exchange critical information, we reduce overhead and accelerate business processes on a global scale.

If you’re an AI Research Engineer at the start of your career and looking for a place to grow fast, learn from experienced colleagues, and work on real AI challenges or someone with a few years of experience ready to level up—Rossum could be the perfect place for you. We’re looking for curious, driven talent who want to tackle real-world AI challenges and push boundaries in a dynamic environment. If that sounds like you, let’s chat!

About the Team

At Rossum, we are building a product-oriented applied AI Research team that works closely with a dedicated Tech Lead and Product Manager to ensure our work delivers maximum value to our users.

The team follows its own agile workflow with scientific method practices at its core: focusing on research, experimentation, iterative development, continuous improvement. Our researchers take end-to-end ownership of their work (with mentorship as needed), with the freedom to experiment, innovate, and contribute to shaping the product.

The team collaborates closely with the AI Platform and Annotation teams to bring AI capabilities into our product. All 3 teams are led by one of Rossum co-founders, Tomáš Tunys.

About our Tech Stack

We pride ourselves on building a modern, scalable, and efficient AI pipeline and fostering forward-thinking engineering culture.

While Python is our main programming language, our environment is intentionally diverse to support cutting-edge research:

• Frameworks & Libraries: PyTorch, PyTorch Lightning, Hugging Face Transformers, TensorFlow, along with good old NumPy, Pandas, and scikit-learn for ML modeling.

• MLOps: Kubeflow Pipelines for training automation, Kubernetes for scalable deployments, Triton Inference Server, ONNX Runtime, vLLM, and more for flexible inference.

• Tooling & Monitoring: JupyterLab, IPyWidgets/Holoviz-based debugging tools, Prometheus, Grafana, and EFK stack for observability.

• Databases & Hosting: PostgreSQL (with pgvector extension for vector similarity), GitLab (code repositories, CI/CD).

You’ll be encouraged to experiment with the latest tools and frameworks to push the boundaries of what our AI platform can do.

About You

We are seeking Junior to Mid-Level AI Research Engineers who have a good foundation in software development and a willingness to learn and apply AI/ML techniques to transform the way the world operates through technology.

Successful candidates typically possess:

• Programming Skills: Basic to intermediate-level proficiency in Python (or other programming languages) and familiarity with AI/ML frameworks (PyTorch, TensorFlow, etc.), as well as software engineering practices like Git and CI/CD.

• CS & Math Foundations: Have a good understanding of algorithms, data structures, optimization, probability, and statistics.

• ML Fundamentals: Familiarity or good understanding of ML fundamentals and the underlying principles (potentially from coursework, internships, or personal projects), and an enthusiasm for deepening these skills.

• Basic MLOps or Cloud Experience: Any experience with cloud-based environments (AWS/GCP/Azure) or orchestration tools (Kubeflow, Airflow, MLFlow) is a plus, though not strictly required.

• Ownership & Innovation: Be proactive, curious, and excited to try new approaches – comfortable receiving guidance from senior team members where needed.

• Communication & Collaboration: General ability and willingness to discuss and share ideas and findings with cross-functional teams and stakeholders.

• Growth Mindset: Willingness to learn from feedback, adopt best practices, and continuously improve.

• Passion for AI Frontier: A genuine desire to contribute to impactful AI solutions and become more knowledgeable in machine learning over time.

About the Role

As a Junior to Mid-Level AI Research Engineer, you’ll work on designing and implementing AI/ML solutions to challenging problems, with mentorship from experienced team members. You’ll play an integral role in bringing AI-powered features to production and influencing how we innovate for our users.

More specifically, you will:

• Contribute to AI projects: Take part in ideation, model prototyping, and deployment under the guidance of senior engineers.

• Collaborate Cross-functionally: Work closely with Product, AI Platform, and Data Annotation teams to identify use cases and deliver business value.

• Learn and Grow with our Platform: Stay up-to-date with AI/ML trends; you’ll be encouraged to experiment with new technologies and integrate them into our platform.

• Focus on Quality & Reliability: Help ensure that the AI solutions are performant, robust, well-monitored, and maintainable.

• Embrace a Learning Culture: Share findings with teammates, contribute ideas in group discussions, and continuously sharpen your AI/ML/engineering skills.

What we offer  

• Cutting-Edge AI & Research – Work on state-of-the-art AI solutions, from deep learning to large-scale document processing.

• End-to-End Impact – Contribute across the full ML lifecycle, from prototyping to production.

• Collaboration with Experts – Join a highly skilled team solving complex AI challenges.

• Fast Iteration & Real-World Application – See your research quickly translate into impactful solutions.

Perks & Benefits

• 5 weeks of vacation + 5 sick days

• Stock options plan

• Flexible hours & hybrid work model

• Extra two weeks of paternity leave

• High-end laptop & tech setup

• English & Czech language lessons

• Multisport card for 2,700+ sports facilities in Czechia