AI Researcher – AI Architecture Research

About the role

Join a research‑driven team in Switzerland focused on advancing AI through innovative model architecture design. You will bridge cutting‑edge theoretical work with practical deployment, publishing high‑impact research while seeing your ideas turned into production‑ready systems.

Key responsibilities

• Research, design, and evaluate novel AI model architectures beyond traditional Transformer approaches.
• Investigate techniques to improve scalability, computational efficiency, training stability, and inference performance.
• Prototype and benchmark new architectures through rigorous experimentation and ablation studies.
• Author and co‑author research papers for top‑tier conferences and journals.
• Collaborate with engineering teams to translate breakthroughs into production‑ready training and inference pipelines.
• Monitor emerging AI advancements and contribute to the long‑term research roadmap.

Required profile

• Strong academic or industrial research background in machine learning, specializing in AI model architecture.
• Demonstrated publication record at venues such as NeurIPS, ICML, ICLR, COLM, ACL, EMNLP, or arXiv.
• Deep understanding of neural network architectures, sequence modeling, attention mechanisms, optimization, and training dynamics.
• Hands‑on experience developing models with PyTorch or JAX.
• Ability to design rigorous experiments, analyze results critically, and communicate findings effectively.
• Comfortable in a fast‑moving, collaborative startup environment with high ownership.

Required skills

• PyTorch
• JAX

What we offer

• Fully remote position providing flexibility and autonomy.