AI Research Engineer - Pre training
Tasks
- Collaborate with engineering and research teams on AI innovation
- Conduct training experiments and analyze results
- Design AI model pre training pipelines across distributed GPUs
- Develop and optimize large language model architectures
- Identify and resolve training performance bottlenecks
- Implement distributed training systems for scalability
- Improve data curation strategies for training outcomes
- Optimize model quality and training efficiency
- Research and prototype multimodal AI systems
Perks/Benefits
- Autonomous work culture
- Flexible work schedule
- Fully remote
- High-performance computing resources
- Professional growth opportunities
Skills/Tech-stack
Data Curation | Deep learning | Distributed Training | GPU Computing | JAX | Machine Learning | Parallel training | PyTorch | Python | TensorFlow | Transformers
Education
Roles
Related jobs
-
API Integration | Anomaly Detection | Data Modeling | Data Pipelines | DockerCareer growth | Flexible work environment | International team collaboration | Remote workMid-level Full TimeSweden R17h ago
-
Accelerate | Deep learning | Diffusers | EEG | FMRIAutonomy | Career growth opportunities | Flexible work environment | Fully remote work | Global collaborationMid-level Full TimeSweden R1d ago
-
AWS | AWS Bedrock | AWS SageMaker | Cloud Computing | Computer VisionFlexible scheduling | Fully remote | Inclusive, diverse team culture | Paid training | Professional development supportMid-level Full TimeSweden R5d ago
-
Machine Learning Engineer SEK 540K-686KAWS | Apache Airflow | CI/CD | Docker | LLMsHybrid work model | Insurance | Paid vacation | Parental leave | PensionMid-level Full TimeStockholm, Sweden R6d ago
-
Senior Data Scientist / AI Engineer - Generative AI SEK 686K-838KAgentic patterns | Evaluation | Experimentation | Generative AI | LLM APIsDiversity and inclusion initiatives | Hybrid working model | Work in English speaking environmentSenior-level Full TimeStockholm, SE, 111 44 R11d ago
-
Senior Data Engineer SEK 700K-838KAWS | Access Control | CI/CD | Data Governance | Data ModelingHack days | Hybrid work | Insurance | Learning culture | Paid vacationSenior-level Full TimeStockholm, Sweden R15d ago
-
Senior Specialist Solutions Architect - AI & ML Engineer SEK 700K-838KAI guardrails | AWS | Agent Orchestration | Agentic Systems | Artificial IntelligenceRemote work | Travel 30 percent maxSenior-level Full TimeFinland; Remote - Denmark; Stockholm, Sweden R26d ago