Granica · 5 days ago
Applied AI Research Engineer – ML Systems & Structured Data
San Francisco Bay Area
Full-time
Onsite
Mid, Senior Level
$160K/yr - $250K/yrGranica is building the next generation of efficient AI infrastructure. The Applied AI Research Engineer will focus on transforming research ideas into practical algorithms and production-ready ML systems that operate across large-scale structured and tabular data.
Artificial Intelligence (AI)Information TechnologySoftware
Responsibilities
Turn research into working systems
Transform foundational ideas from Granica Research and Prof. Andrea Montanari’s group into scalable algorithms and prototypes
Build evaluation harnesses, datasets, and benchmarks that measure real signal from research ideas
Define and improve metrics that quantify progress in structured AI systems
Invent and optimize algorithms
Develop efficient learning methods for relational, tabular, graph, and enterprise datasets
Prototype representation learning architectures and compression-aware models
Explore new approaches for learning from heterogeneous structured data
Build high-performance ML pipelines
Implement fast training and inference pipelines using PyTorch, JAX, or custom kernels
Optimize memory usage, compute utilization, and data movement
Improve cost, latency, and throughput for large-scale ML workloads
Build hybrid AI systems
Design systems integrating symbolic, relational, and neural components
Enable AI models to reason over structured datasets without relying on text intermediaries
Collaborate across research and engineering
Work with Research Scientists to validate hypotheses at scale
Work with Systems Engineers to integrate algorithms into Granica’s data platform
Work with Product Engineering to ship features powering real enterprise workloads
Iterate fast and measure everything
Run controlled experiments and analyze performance improvements
Deliver results with clear benchmarks and reproducible evaluations
Drive the cycle from prototype → production → optimization
Qualification
Required
Strong background in machine learning, probabilistic modeling, optimization, or large-scale ML systems
Experience building algorithms for structured, relational, tabular, or graph data
Ability to reason from first principles about scaling behavior, efficiency, and information flow
Hands-on experience with PyTorch, JAX, TensorFlow, or similar ML frameworks
Strong programming skills in Python
Experience with systems languages such as Rust, C++, or CUDA is a plus
Experience building large-scale ML pipelines, evaluation frameworks, or distributed systems
Proven ability to turn research ideas into performant, reliable code
Comfort working in research-driven environments with ambiguous problem definitions
Strong experimentation discipline and focus on measurable performance improvements
Preferred
Structured representation learning, tabular ML, relational learning, or graph ML
Experience with large-scale training infrastructure or distributed ML
Familiarity with data systems, query engines, or large-scale data pipelines
Experience building evaluation infrastructure for ML systems
Open-source contributions or collaborative work bridging research and production systems
Benefits
Meaningful equity
Substantial bonus for top performers
Flexible time off
Comprehensive health coverage for you and your family
Support for research, publication, and deep technical exploration
Company
Granica
Granica is the world's first AI Data Readiness Platform, creating cutting-edge and enterprise-ready AI infrastructure services.
Founded in 2019
11-50 employees
H1B Sponsorship
Granica has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2026 (3)
2025 (4)
2024 (7)
2023 (2)
Funding
Current Stage
Early StageTotal Funding
$45MKey Investors
New Enterprise Associates
2023-06-08Series A· $45M
2020-01-01Seed
Leadership Team
Rahul Ponnala
CEO & Co-Founder
Recent News
Google Patent
2025-02-07
Company data provided by crunchbase