Pytho AI Logo

Pytho AI

AI Engineer

Reposted 2 Months Ago
In-Office
San Francisco, CA, USA
Junior
In-Office
San Francisco, CA, USA
Junior
As an AI Engineer, you will design, train, and deploy machine learning models, build data pipelines, and integrate models into production systems while collaborating with the engineering team on advanced AI solutions.
The summary above was generated by AI
Company Overview

Prepare first. Decide first. Win first.

At Pytho, we're on a mission to ensure warfighters are prepared for every challenge they face. We are committed to assembling a world-class team of exceptional problem solvers, combining expert engineering talent with deep experience in National Security.

We build AI for the most important problem in defense — making sure the force is ready to fight. Our software is deployed and in use with DoW partners today.

The Opportunity

We're building out our AI Engineering Team. This is a chance to join at the earliest stage of a company where the engineering problems are genuinely hard — planning under uncertainty, multi-agent coordination, simulation, reinforcement learning — and the work is deployed in environments where the output actually matters.

You will work across the full ML lifecycle — transforming messy, real-world data into production-grade models deployed in high-stakes environments. You'll ship fast, work directly with the founders, and have outsized influence on the technical direction of the company.

Key Responsibilities

AI/ML Development: Design, train, fine-tune, and deploy ML models to solve real-world operational problems. Own the lifecycle from data exploration and feature engineering to evaluation and production monitoring.

Data & Systems Engineering: Build scalable data pipelines for structured and unstructured data (text, imagery, geospatial, sensor data). Develop reliable training and inference systems optimized for performance and edge deployment.

Engineering Excellence: Integrate models into robust, scalable production systems with strong testing, observability, and CI/CD practices.

Research & Experimentation: Prototype and benchmark new modeling approaches (LLMs, multimodal systems) to improve performance, robustness, and mission impact.

What You Bring

Required

  • Bachelor's degree (B.Sc.) in Computer Science

  • Strong software engineering fundamentals

  • Solid understanding of machine learning principles (model evaluation, optimization, bias/variance tradeoffs)

  • Hands-on experience with ML frameworks (PyTorch, TensorFlow, Hugging Face)

  • Experience working with real-world datasets — cleaning, feature engineering, experimentation, and performance analysis

  • Strong communication and collaboration skills

  • Willingness to travel up to 15% to engage with DoW partners

  • Must be eligible to obtain and maintain a U.S. security clearance

Preferred

  • Master's degree (M.Sc.) in Computer Science, Machine Learning, or related field

  • Experience with edge-deployable or offline ML systems

  • Experience optimizing models for latency and compute constraints

  • Familiarity with distributed training or large-scale data processing

  • Exposure to geospatial, multimodal, or reasoning systems

  • Experience with containerization and orchestration tools

Benefits
  • Competitive salary and equity

  • Opportunity to help build a category-defining defense tech company

  • Full health benefits

  • 401(k)

  • Offices in Washington DC and San Francisco

Similar Jobs

2 Hours Ago
Hybrid
San Jose, CA, USA
197K-246K Annually
Senior level
197K-246K Annually
Senior level
Fintech • Machine Learning • Payments • Software • Financial Services
Build and operate foundational AI systems, including foundation-model training, LLM inference, agentic workflows, similarity search, guardrails, evaluation, governance, and observability. Lead architecture, optimization, reliability, and deployment of scalable AI platforms; improve GPU/TPU utilization, latency, throughput, and cost. Partner across engineering, research, product, and infrastructure teams, guide technical strategy, conduct deployment reviews, and mentor senior technical staff.
Top Skills: AWSAws UltraclustersAzureC#C++CudaGoGCPHugging FaceJavaPythonPyTorchScalaTpuVectordbs
2 Hours Ago
Hybrid
2 Locations
197K-246K Annually
Mid level
197K-246K Annually
Mid level
Fintech • Machine Learning • Payments • Software • Financial Services
Build and deploy scalable generative AI platform services, including foundation model training, LLM inference, agentic workflows, vector search, guardrails, evaluation, governance, and observability. Optimize AI systems for cost, latency, throughput, reliability, and hardware utilization. Own complex system architecture, define reliability objectives, lead technical reviews, ensure security and compliance, and mentor senior technical staff.
Top Skills: AWSAws UltraclustersAzureC#C++CudaDistributed SystemsGoGoGCPGpuHugging FaceJavaLarge Language ModelsPythonPyTorchScalaSimilarity SearchTpuVector Databases
2 Hours Ago
Hybrid
San Jose, CA, USA
230K-286K Annually
Senior level
230K-286K Annually
Senior level
Fintech • Machine Learning • Payments • Software • Financial Services
Design, develop, deploy, and optimize production AI systems, including foundation models, LLM inference, agentic workflows, vector search, guardrails, and model evaluation. Lead orchestration, cost-performance governance, technical standards, design reviews, and roadmap development. Apply advanced optimization techniques to improve scalability, latency, throughput, hardware utilization, and cost efficiency while mentoring senior technical staff and ensuring responsible AI practices.
Top Skills: AWSAws UltraclustersAzureC#C++CudaGoGCPHugging FaceJavaPythonPyTorchScalaVectordbs

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account