Machine Learning PhD student at Oxford University
Previously PhD Resident at Google X
Co-author of Optimistix: modular nonlinear optimisation in JAX.
Co-author of Lineax: linear solvers and least-squares in JAX.
Lineax is also available on arXiv.
@dataclass
class About:
location: Any
research: ScientificML = ["Optimisation", "Numerical Methods", "Deep Learning"]
main_tools: Python = ["JAX", "Equinox"]