Edward Jex

MEng Computer Science, Durham 2025, top of the year. I build small, efficient models: a 12,387-weight detector that runs on 2.2 µJ a frame, 680× less than YOLO11-nano; a 227k-parameter model that reads strikethrough in patent amendments at 3 ms a line on CPU; and a diffusion model in under a million parameters. At Solve Intelligence (YC S23), a fast-moving legal AI startup, I also rebuilt the agent loop as a service that runs thousands of LLM agents in the background. My research interests are mechanistic interpretability and reinforcement learning.

Experience

Software Engineer

July 2025–present

Solve Intelligence (YC S23)

  • Rewrote the core agent loop as its own microservice, out of the API. Agents now run in the background for hours, survive deployments, and scale to thousands at once. Results stream over Redis and server-sent events to the frontend, where they trigger actions; a reconnecting client resumes from a cursor with no gaps or duplicates.
  • Took Charts from prototype to a production product used by thousands of customers, working end to end: the agent loop and AI tools, a planner delegating to subagents, through to the rich-text, cell-based editor. Tens of hours of attorney work now take minutes.
  • Scaled agents from short, bounded tasks to long, open-ended ones: lazy document loading and parallel OCR for time to first token, and hunting down GIL-bound work to fit many agents on one container.
  • Shipped across the stack: Postgres row-level security, live sharing, and patent language and jurisdiction conversion.
  • Trained a 227,746-parameter (834 KB) model that labels each OCR character as struck or underlined, at 3 ms a line on two CPU threads, replacing image tokens at a sixth of the context cost. Took it apart with a pre-registered, 76-run retraining ablation: a placebo set the noise floor at a quarter of a point, and removing a part and retraining cost far less than switching it off (3.7 points against 94).

Research and projects

Energy-efficient object detection on edge devices

MEng thesis, supervised by Dr Farshad Arvin

Asked how little energy a useful detector can spend. Treated detection as one heatmap peak per animal rather than boxes, trained five hundred architectures across a GPU cluster and measured each on a Hailo-8L, because architecture explained only 15% of the variance in energy on the accelerator (80% on CPU). The best has three layers and 12,387 weights, reaching 0.948 AP at 2.2 µJ a frame, 680× less than YOLO11-nano. It runs in the browser on hand-written WGSL compute shaders, with every layer's feature maps inspectable.

Deep learning under a parameter budget

Module 90%, 2024

A class-conditional diffusion model with classifier-free guidance in under a million parameters, reaching FID 18.1 on CIFAR-100, and a 100k-parameter MobileNetV3 variant with grouped pointwise convolutions.

Sample-efficient RL for BipedalWalker

Module 90%, 2024

Compared four model-free methods, then extended TD-MPC2 to solve BipedalWalker in under 100 episodes.

Temporally consistent video translation

Module 75%, 2025

A UVCGAN2-style ViT-UNet CycleGAN, extended with MoCycle-GAN motion losses from RAFT optical flow, trained on an A100.

Education

Durham University

2021–2025

MEng Computer Science, First Class

81% four-year average, 86% in the final year; prizes for the top project (91%) and the top overall grade. Data Compression 93%, Quantum Computing 89%, Advanced Algorithms 88%.

A Levels: 4 A*s

2021

Maths, Further Maths, Physics, Computer Science.

Skills

Machine learning

  • Hardware-aware NAS, quantisation, energy measurement
  • Edge deployment: ONNX, Hailo, WebGPU
  • PyTorch, JAX, Flax; GPU clusters
  • CNNs, transformers, diffusion, GANs
  • RL: TD-MPC2, SAC, REDQ, DropQ, CrossQ; MCTS

Engineering

  • Python, Rust, TypeScript, C/C++
  • Production LLM agents: orchestration, tool use, streaming
  • Postgres, Redis, Celery, FastAPI, React
  • AWS, GCP, Docker, CI/CD

Other experience

Web Development Intern

Summer 2022

Cisco, Webex web client

Rich message rendering, build system migration, React and Redux UI components.

Teaching Assistant

2023–2025

Durham Computer Science

Ran labs for Computational Thinking and Networks and Systems.

River level forecasting

2024

PyTorch forecast served from Rust with ONNX Runtime, used by Durham rowing clubs. river.edjex.dev

President

2023–2024

Trevelyan College Boat Club

Ran finances, training and racing; replaced ageing equipment and grew the membership.