A1
Founding AI/ML Research Engineer(m/w/x)
Vollzeit
Remote
Senior
Keine Angabe
Berlin
A1
A116 Tage
Founding Lead Machine Learning Engineer(m/w/x)
Vollzeit
Remote
Senior
AI/ML Job
Keine Angabe
Berlin
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 high-performance 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 state-of-the-art 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
- •Prototype, test, iterate, and ship solutions
- •Collaborate with a small, senior team on major decisions
- •Lift the bar for yourself and your teammates
Tools & Technologien
PyTorchJAXTensorFlowDeepSpeedFSDPMegatronZeRORayvLLMTensorRT-LLMFasterTransformerApache ArrowSpark
Sprachen
Englisch – verhandlungssicher
Deine Vorteile
Attraktive Vergütung
- •Competitive compensation
Boni & Prämien
- •Performance-based bonuses
Flexibles Arbeiten
- •Remote-first flexibility
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?