A1
Founding Machine Learning Engineer(m/w/x)
Vollzeit
Remote
Berufserfahren
Keine Angabe
Zürich
A1
A13 Tage
Founding Lead Machine Learning Engineer(m/w/x)
Vollzeit
Remote
Senior
AI/ML Job
Keine Angabe
Zürich
Nejo KI-Zusammenfassung
In this founding role, you will drive the technical vision for A1, designing innovative systems and experimenting with cutting-edge models. Your work will directly shape a new global AI product, collaborating closely with a small, elite team to tackle meaningful challenges.
Lass KI die perfekten Jobs für dich finden!
Lade deinen CV hoch und die Nejo-KI findet passende Stellenangebote für dich.
Anforderungen
- •Strong background in deep learning and transformer architectures
- •Hands-on experience training or fine-tuning large models (LLMs or vision models)
- •Proficiency with PyTorch, JAX, or TensorFlow
- •Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)
- •Strong software engineering skills — writing robust, production-grade systems
- •Experience with GPU optimization: memory efficiency, quantization, mixed precision
- •Comfortable owning ambiguous, zero-to-one technical problems end-to-end
- •Experience with LLM inference frameworks (vLLM, TensorRT-LLM, FasterTransformer)
- •Contributions to open-source ML libraries
- •Background in scientific computing, compilers, or GPU kernels
- •Experience with RLHF pipelines (PPO, DPO, ORPO)
- •Experience training or deploying multimodal or diffusion models
- •Experience in large-scale data processing (Apache Arrow, Spark, Ray)
- •Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI)
Keine Angabe
Berufserfahrung
ca. 4 – 6 Jahre
Deine Aufgaben
- •Shape the core technical direction of A1
- •Select models and define training strategies
- •Design infrastructure and long-term architecture
- •Create end-to-end training pipelines
- •Build evaluation frameworks for performance and safety
- •Architect scalable inference systems
- •Fine-tune models using advanced methods
- •Develop data systems for high-quality training data
- •Implement alignment, safety, and guardrail strategies
- •Own deployment processes, including GPU optimization
- •Experiment with new use cases and AI-powered experiences
- •Explore frontier techniques like retrieval-augmented training
- •Adapt open-source frontier models to new architectures
- •Prototype, test, iterate, and ship solutions
- •Collaborate with a small, high-performance team
- •Set engineering standards and contribute to team culture
Tools & Technologien
PyTorchJAXTensorFlowDeepSpeedFSDPMegatronZeRORayvLLMTensorRT-LLMFasterTransformerApache ArrowSpark
Sprachen
Englisch – verhandlungssicher
Deine Vorteile
Attraktive Vergütung
- •Competitive compensation
Boni & Prämien
- •Performance-based bonuses
Sonstige Vorteile
- •Insurance coverage
- •Global travel insurance
Workation & Sabbatical
- •Flexible time off
Die Originalanzeige dieses Stellenangebotes in der aktuellsten Version findest du hier. Nejo hat diesen Job automatisch von der Website des Unternehmens A1 erfasst und die Informationen auf Nejo mit Hilfe von KI für dich aufbereitet. Trotz sorgfältiger Analyse können einzelne Informationen unvollständig oder ungenau sein. Bitte prüfe immer alle Angaben in der Originalanzeige! Inhalte und Urheberrechte der Originalanzeige liegen beim ausschreibenden Unternehmen.
Ähnliche Jobs direkt in deine Inbox?
Noch nicht perfekt?