Software Engineer at Graphon.AI

Compensation

$180,000 – $250,000/yr

Location: San Francisco, California
Type: Full-time
Level: not_specified
Posted: May 14, 2026

Description

Graphon AI is seeking a highly skilled individual to join their team as a Software Engineer specializing in machine learning infrastructure. The successful candidate will contribute to building the pre-model intelligence layer for AI, focusing on creating scalable memory and context solutions using a graph-based approach. This role involves strong software engineering fundamentals, experience with Python, and familiarity with modern ML stacks like PyTorch/JAX and distributed training.

Required Skills

software engineering fundamentals
building production systems
Python
C++/Go
machine learning infrastructure
training pipelines
inference/serving systems
data pipelines
model deployment
modern ML stacks
PyTorch/JAX
GPUs
distributed training or inference
core systems concepts
version control
Linux
basic networking
CI/CD

Additional Information

Required Previous Experiences: experience building production systems (Python required; experience with C++/Go a plus)
Experience working with **machine learning infrastructure**, such as training pipelines, inference/serving systems, data pipelines, or model deployment.
Familiarity with modern ML stacks (e.g., PyTorch/JAX, GPUs, distributed training or inference).
Solid understanding of core systems concepts (version control, Linux, basic networking, CI/CD).
Preferred Qualifications: Experience scaling ML systems in production environments.
Experience with distributed systems, large-scale data processing, or GPU optimization.
Exposure to multimodal ML, retrieval systems, or graph-based representations.
Prior startup experience or comfort operating in ambiguous, fast-moving environments.