Agilesoft · 2 months ago
Research Engineer - Applied AI
San Francisco Bay Area
Full-time
Onsite
Mid, Senior Level
$160K/yr - $250K/yrAgilesoft is an AI research and systems company focused on building infrastructure for structured and efficient intelligence. The Applied AI Research Team is seeking a Research Engineer to transform research ideas into practical algorithms and production-ready systems, working with large-scale structured data to enhance enterprise AI capabilities.
Information Technology & Services
Responsibilities
Transform foundational ideas from Granica Research and Prof. Andrea Montanari’s group into scalable algorithms and experimental prototypes
Build the evaluation harnesses, metrics, and datasets that reveal real signal from research concepts
Define and refine the metrics that determine progress in structured AI
Develop efficient learning methods for relational, tabular, graph, and enterprise data
Prototype representation learning architectures and compression-aware models for large-scale structured information
Implement fast training and inference loops using PyTorch, JAX, or custom kernels
Optimize memory, compute, and data-movement paths with a focus on cost, latency, and throughput
Design hybrid learning systems that reason over structured data natively, not through text intermediaries
Work with Research Scientists to validate hypotheses at scale
Work with Systems Engineers to integrate your algorithms into Granica’s core data platform
Work with Product Engineering to ship features that power live enterprise workloads
Run controlled experiments, analyze performance deltas, and deliver results with clear benchmarks
Drive the loop from prototype to production, improving the system each cycle
Qualification
Required
Strong background in machine learning, probabilistic modeling, optimization, or distributed learning
Experience building and tuning algorithms for structured, tabular, graph, or relational data
Ability to reason from first principles about efficiency, scaling behavior, and information flow
Hands-on experience with PyTorch, JAX, TensorFlow, or custom ML kernels
Strong Python skills and familiarity with systems languages such as Rust, C++, or CUDA
Experience with large-scale data pipelines, model evaluation frameworks, or distributed systems
Demonstrated ability to turn theoretical concepts into performant, reliable code
Comfort working in ambiguous research environments while delivering measurable outcomes
Curiosity for how structure and efficiency reshape the next generation of AI
Preferred
Structured representation learning, tabular or multimodal models, or relational ML
Distributed data systems, query engines, or vector/embedding infrastructure
Open-source contributions or collaborative research bridging theory and production
Benefits
Flexible time off plus comprehensive health coverage for you and your family
Support for research, publication, and deep technical exploration
Company
Agilesoft
With nearly twenty years of experience, we specialize in providing comprehensive web solutions.
Founded in 2006
2-10 employees
H1B Sponsorship
Agilesoft 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
2025 (13)
2024 (15)
2023 (5)
2022 (10)
2021 (4)
2020 (12)
Funding
Current Stage
Early StageCompany data provided by crunchbase