2025
u-μP: The Unit-Scaled Maximal Update Parametrization.
Proceedings of the Thirteenth International Conference on Learning Representations, 2025

2024
Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasets.
Proceedings of the Twelfth International Conference on Learning Representations, 2024

HUNIPU: Efficient Hungarian Algorithm on IPUs.
Proceedings of the 40th International Conference on Data Engineering, ICDE 2024, 2024

2023
GPS++: Reviving the Art of Message Passing for Molecular Property Prediction.
Trans. Mach. Learn. Res., 2023

Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasets.
CoRR, 2023

Generating QM1B with PySCF<sub>IPU</sub>.
Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

2022
GPS++: An Optimised Hybrid MPNN/Transformer for Molecular Property Prediction.
CoRR, 2022