As a Machine Learning Engineer, you will build models and agents to accelerate scientific discovery, collaborating with researchers and developing efficient infrastructure.
About
Edison Scientific builds and commercializes AI agents for science. Scientific discovery moves too slowly, and autonomous AI agents are how we intend to fix that. We're assembling a team of top researchers and engineers across AI and biology to build an AI scientist.
RoleAs a Principal Machine Learning Engineer at Edison Scientific, you play a central role in building the models and agents that accelerate scientific discovery. You will work on both cutting edge research and practical engineering, bridging advanced machine learning concepts with robust, reliable software that real scientists depend on.
This role is on-site at our San Francisco office in the Dogpatch neighborhood. Our office is a converted warehouse with high ceilings, open space, and a team excited about what we’re building.
Responsibilities- Interpret qualitative challenges in building AI agents for science as well-formulated optimizable problems
- Build appropriate environments in which to train and deploy AI agents that solve scientific tasks
- Work with scientists to formulate training data pipelines, and scale them, ensuring observability and reproducibility
- Lead training of large-scale LLM-based systems, including building internal infrastructure to improve the efficiency of experimentation and production training runs
- Build efficient and flexible inference infrastructure, supporting complex sampling algorithms and custom architectures
- Develop and extend our experimentation platform for internal tools and projects.
- Collaborate closely with a multidisciplinary team of AI researchers, chemists, biologists, fostering an environment of innovation and discovery.
- 8-10+ years of strong track record of work in applied ML research and application of ML methods to solving real-world problems
- Experience working across the ML lifecycle: data pipelines and provenance, model training, model deployment, and validation in production systems.
- Fluency in PyTorch, Jax or equivalent framework.
- Demonstrated experience with experimentation in academic or industry settings.
- Strong programming expertise with the capability to adapt to various technical challenges in the data, ML, and LLM software stack.
- PhD in Machine Learning, Computer Science, or other quantitative field
- Familiarity with leveraging and managing distributed computing resources
- Background architecting complex distributed systems
Salary
$275,000 - $350,000 • Offers equity
Why join us?
- Competitive salary and equity
- Full healthcare coverage — we pay 100% of premiums for you and your dependents
- Support for growing families, including a yearly new parent stipend and fertility coverage through Carrot
- 401(k) company matching
- $300 health and wellness benefit
- Lunch is on us every day you're in the office, and dinner is on us when you're working late
- Regular team offsites and company events
- A fast-moving, mission-driven culture where smart people do their best work and actually enjoy doing it
Edison Scientific San Francisco, California, USA Office
San Francisco, California, United States, 94107
Similar Jobs
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Leads the technical vision and architecture for AI-powered exploitability analysis in cybersecurity. Defines advanced modeling approaches, enterprise-scale engineering standards, AI safety and governance practices, and portfolio-wide architecture. Guides multiple teams without direct authority, mentors senior engineers, represents technology to executives and customers, and delivers production AI/ML products using LLMs, agents, retrieval, probabilistic scoring, and graph technologies.
Top Skills:
Agentic ArchitecturesAi FrameworksAi InfrastructureAPIsClaude CodeCloud-Native PlatformsCodexCursorData SystemsDistributed SystemsFoundation ModelsGoGraph AnalyticsGraph SystemsJavaLlmsMicroservicesModel EvaluationModel ObservabilityProbabilistic ModelingPythonRetrieval SystemsRetrieval-Augmented Generation (Rag)Vector TechnologiesWindsurf
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Lead end-to-end machine learning development for autonomous driving, including data curation, foundation-model pre-training, post-training, fine-tuning, alignment, and deployment. Build scalable pipelines using real and synthetic datasets, apply imitation and reinforcement learning, improve model safety and reliability, collaborate across teams, lead technical initiatives, and mentor ML engineers.
Top Skills:
SparkNumpyPandasPythonPyTorch
Security
Provides AI/ML systems engineering and technical advisory support for an NRO space launch program. Responsibilities include developing AI/ML strategies and roadmaps, guiding model and algorithm design, translating mission needs into AI/ML opportunities, optimizing data pipelines and cloud environments, validating model performance, and mentoring practitioners. Requires extensive AI/ML experience, cloud and MLOps expertise, strong stakeholder communication, SECURE Plus certification, and an active Top Secret clearance with polygraph.
Top Skills:
Ai/MlAWSAzureCi/CdDevOpsGCPMlopsPythonPyTorchRTensorFlow
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
