Senior ML Engineer (Token Factory)
Tasks
- Analyze sharding strategies and custom kernels
- Apply software engineering best practices
- Build JAX training and evaluation pipelines
- Develop fine tuning methodologies
- Develop scaling laws
- Experiment with model architectures
- Implement low precision training and inference
- Improve large language model inference performance
- Optimize model quality and training efficiency
- Perform distributed training across nodes
- Translate research into production systems
Perks/Benefits
- Career growth
- Collaborative team
- Continuous learning
- Flexible working hours
- High ownership
- Inclusive workplace
- International work environment
- Support during hiring process
Skills/Tech-stack
Autoregressive models | CI/CD | Custom Kernels | Deep learning | Distributed Training | FP8 | Fine Tuning | GPU Computing | Inference Optimization | JAX | Language Models | Large Language Models | LoRA | Low Precision | Low-precision computing | Machine Learning | Mixture of Experts | Parallel Computing | Precision computing | Python | Quantization | Reinforcement Learning | Sharding | Speculative decoding | Test automation | Unit Testing | Version control
Education
N/A
Related jobs
-
Azure Data | Azure Data Factory | Azure Data Lake | Azure Data Lake Storage | Azure Data Lake Storage Gen2Digital transformation project exposure | Fully remote | Opportunity to use AI powered engineering tools | Remote flexibility | Work with modern Microsoft Azure data ecosystemSenior-level Contract Full TimeSpain R1d ago
-
APIs | AWS Lambda | AWS S3 | Amazon RDS | Amazon SNSAutonomy | Collaborative team | Flexible location | Fully remote | OwnershipSenior-level Full TimeSpain R1d ago
-
AWS | Async/Await | Asynchronous programming | Azure | CI/CDAsynchronous Working Environment | Career development | Hardware provided | Hybrid arrangements | Opportunity to work abroadSenior-level Full TimeSpain R1d ago
-
Expert Data Science EUR 35K-65KAgentic artificial intelligence | Artificial Intelligence | Data Governance | Data Quality | Data VisualizationGlobal recognition programs | Hybrid working | Insurance plans | Paid parental leave | Relocation support not offeredSenior-level Full TimeBarcelona Gran Vía, Spain R1d ago
-
Expert Data Science EUR 35K-65KAgentic artificial intelligence | Artificial Intelligence | Data Governance | Data Quality | Data ScienceHybrid working | Insurance plans | Paid parental leave | Relocation support none | Retirement plansSenior-level Full TimeBarcelona Gran Vía, Spain R1d ago
-
Forward Deployed AI Engineer, Spain - BCG X EUR 29K-29KAmazon Web Services | Apache Spark | Azure | Cloud platform | DaskEntry-level Full TimeMadrid, Spain1d ago
-
Algorithms | Computational Efficiency | Data analytics | Machine Learning | PythonEntry-level InternshipMadrid, Spain1d ago
-
Data Engineer EUR 48K-60KAmazon DynamoDB | Amazon SageMaker | Amazon Web Services | Apache Spark | C#Flexible working model | Health resources | Holiday and leave allowance | Hybrid work model | Meal voucherMid-level Full TimeBarcelona R1d ago
-
Embedded WiFi Technical Lead EUR 60K-75KARM | ARM toolchains | Access Point | Access Point Mode | Analog electronicsBeta Testing Program | Discounts on Ajax products | Medical insurance | Partner company discountsSenior-level Full TimeBarcelona, Catalonia1d ago
-
Algorithms | Cybersecurity | Data Processing | Data Structures | DebuggingSenior-level Full TimeMálaga, Spain1d ago
-
Data Analyst/UX Engineer- Ensayos en avión M/F EUR 50K-63KAngular | Machine Learning | Matplotlib | MySQL | NumPyMid-level ContractEurope, Spain1d ago
-
Senior Data Engineer EUR 70K-90KApache Kafka | CI/CD | ClickHouse | Docker | ElasticsearchCoaching and training | Flexible working hours | Learning and development | Remote-first | Six dedicated learning days per yearSenior-level Full TimeGermany /Spain / Romania (Remote) R2d ago
-
Staff AI Software Engineer EUR 80K-120KAI Agents | AWS | Agentic AI | Amazon EKS | AutogenCompetitive benefits | Learning opportunities | Work-life balanceSenior-level Full TimeMadrid Office2d ago
-
Data Engineer / Architect EUR 54K-79KAWS | Apache Spark | Azure | Big Data | CI/CDBirthday day off | English lessons | Flexible schedule | Flexible work from home | Free Microsoft certificationsSenior-level Full TimeSpain - Remote R2d ago
-
AI Software Engineer | Spain EUR 40K-62KAPI Integration | Agent Orchestration | Anthropic API | Braintrust | CI/CDClient embedding workshops proofs of concept code with sessions architecture walkthroughs | Pathway to Forward Deployed Engineer programme | Vendor fellowship accessSenior-level Full TimeMadrid, Torre Chamartin, Spain2d ago
-
AWS | AWS AgentCore | AWS Bedrock | AWS Lambda | AWS StrandsSenior-level Full TimeBarcelona, Catalonia, ESP2d ago
-
AWS | AWS AgentCore | AWS Bedrock | AWS Strands | Amazon QuickSightSenior-level Full TimeBarcelona, Catalonia, ESP2d ago
-
Data Engineer EUR 20K-23KApache Spark | Big Data | Data Analysis | Data Architecture | Data ModelingContinuous training | Flexible schedule | Hybrid work model | Life and accident insuranceEntry-level Full TimeSpain R2d ago
-
AWS | Apache Kafka | CI/CD | Docker | ELK100 percent remote work | Collaborative culture | Company offsite weeks | Complex digital health projects | Flexible work locationSenior-level Full TimeSpain R2d ago
-
AI integration | AWS | Agentic Workflows | Azure | Best practicesFlexible contract engagement | Global work flexibility | High-impact workEntry-level Contract Full TimeSpain2d ago
-
Data Architect EUR 54K-75KAPIs | AWS | Analytical Data | Analytical Data Platforms | Apache AirflowSenior-level Full TimeValencia, Spain2d ago
-
AI Architect (iGaming) USD 160K-209KAgent systems | Agentic AI | Backend Development | Client development | Game Client DevelopmentFlexible allocation | Freelance contract | Knowledge transfer sessions | Pilot ProjectSenior-level Full TimePoland R2d ago
-
Machine Learning Engineer EUR 62K-80KClassification | Dashboards | Data Warehousing | Git | Hypothesis TestingMid-level Full TimeAmsterdam, Netherlands, sennder HQ Berlin, Barcelona, …2d ago
-
AI Engineering Workflow Developer (mf/d) EUR 40K-60KCI/CD | Docker | FastAPI | Git | GitHub API26 plus working days vacation | Childcare support | English and German language classes | Flexible working hours | Free access to specialist servicesEntry-level Full TimeGranada, AN, Spain R2d ago
-
Alerting | Data Pipelines | Distributed Systems | JAX | LoggingMid-level Full TimeSpain R3d ago