Machine Learning Engineer
London/Cambridge - Hybrid
We're partnered with a start-up who are developing a ground-breaking architecture for logical reasoning in AI, tackling real-world challenges in semiconductors, quantum computing, and more.
Responsibilities:
- Design and develop novel machine learning models for theorem proving and logical reasoning in AI.
- Utilise supervised, unsupervised, and reinforcement learning techniques.
- Preprocess, clean, and transform large datasets for model training and evaluation.
- Implement data pipelines and ETL processes for efficient data flow.
- Conduct experiments, analyse results, and optimise machine learning model performance.
Required Experience:
- Large scale model training, multi-GPU training.
- Training large models with over 20B parameters.
- Deep theoretical understanding of the transformers design and implementation.
- Programming Language Processing, training models for code generation.
- Development of autonomous software development.
- Ideally worked in a start-up.
- Ideally has a PhD in Computer Science, Mathematics or a similar field.
Nice to have:
- Jax, MegaTron-LM experience,
- Knowledge of alternative architectures including the GNN, Mamba, RNNx,…
- Publications of the journal papers in the top conferences.
- Active in developing open source or passion projects and sharing them on GitHub, Medium or similar.
NOTE: We are unable to offer sponsorship and candidates must be based in the UK.