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.

DigiTech Resourcing jobs

Machine Learning Engineer