AI Researcher — AI Architecture Research at Jobgether · Sweden

This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI Researcher — AI Architecture Research based in Sweden.

Join a research-driven team dedicated to advancing the next generation of artificial intelligence through innovative model architecture design. In this role, you'll explore novel approaches to machine learning, bridging cutting-edge theoretical research with practical deployment in production environments. You'll collaborate with experienced researchers and engineers to develop, evaluate, and optimize new AI architectures that push the boundaries of performance, scalability, and efficiency. This is an exciting opportunity for researchers who are passionate about publishing impactful work while seeing their ideas translated into real-world AI systems. If you thrive in a highly technical, fast-paced environment and enjoy tackling fundamental AI challenges, this role offers exceptional scope for innovation and professional growth.

Accountabilities

• Research, design, and evaluate novel AI model architectures beyond traditional Transformer-based approaches.
  • Investigate new techniques to improve model scalability, computational efficiency, training stability, and inference performance.
  • Prototype innovative architectures and validate their effectiveness through rigorous experimentation, benchmarking, and ablation studies.
  • Author and co-author high-quality research papers for leading machine learning conferences, journals, and preprint platforms.
  • Collaborate closely with engineering teams to translate research breakthroughs into production-ready training and inference systems.
  • Monitor emerging advancements in artificial intelligence research and identify promising directions for future innovation.
  • Contribute to defining the long-term research roadmap and influence the evolution of next-generation AI models.

Requirements

    - Strong academic or industrial research background in machine learning, with a specialization in AI model architecture.
    - Demonstrated publication record in recognized AI or machine learning venues such as NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, or arXiv.
    - Deep understanding of neural network architectures, sequence modeling, attention mechanisms, optimization techniques, and training dynamics.
    - Hands-on experience developing machine learning models using PyTorch or JAX.
    - Proven ability to design rigorous experiments, analyze results critically, and communicate research findings effectively.
    - Comfortable working in a fast-moving, collaborative startup environment with a high level of ownership.
    - Experience with non-Transformer architectures, long-context models, state-space models, hybrid architectures, or memory-efficient AI systems is highly valued.
    - Familiarity with efficient training techniques, inference optimization, open-source research contributions, or production AI systems is considered an advantage.

Benefits

        - Fully remote position offering flexibility and autonomy.
        - Competitive compensation package with meaningful equity participation.
        - Opportunity to define research direction and influence next-generation AI architectures.
        - Strong support for publishing impactful research in leading conferences and journals.
        - Close collaboration between research and engineering, enabling rapid deployment of new ideas.
        - Highly technical, research-focused team with a collaborative and innovative culture.
        - Exposure to challenging AI problems with real-world impact and significant opportunities for professional growth.

How Jobgether works:

We use an AI-powered matching process to ensure your application is reviewed quickly, objectively, and fairly against the role's core requirements. Our system identifies the top-fitting candidates, and this shortlist is then shared directly with the hiring company. The final decision and next steps (interviews, assessments) are managed by their internal team.

We appreciate your interest and wish you the best!

Why Apply Through Jobgether?

Data Privacy Notice: By submitting your application, you acknowledge that Jobgether will process your personal data to evaluate your candidacy and share relevant information with the hiring employer. This processing is based on legitimate interest and pre-contractual measures under applicable data protection laws (including GDPR). You may exercise your rights (access, rectification, erasure, objection) at any time.

#LI-CL1