Easy Apply
Easy Apply
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
Top Skills
Jax
PyTorch
Edison Scientific San Francisco, California, USA Office
San Francisco, California, United States, 94107
Similar Jobs
Automotive • Big Data • Insurance • Software • Transportation
The Principal ML Engineer at Agero will develop and productionize a Dispatch System using ML models, optimize decision-making processes, and lead a team to improve operational performance.
Top Skills:
AirflowAWSAzureGCPPythonPyTorchSagemakerSQLXgboost
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Lead end-to-end AI/ML projects: research, prototype, optimize, and productionize ML and LLM solutions for healthcare. Mentor and recruit teams, run experiments and A/B tests, present results to leadership, and pursue AI research and IP.
Top Skills:
Deep LearningDpoGansGenerative AiGnnsLarge Language Models (Llms)MultimodalNlpPpoPythonPyTorchSsmsTensorFlowTransformers
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Lead design, architecture, and deployment of end-to-end ML solutions for onboard autonomous driving. Apply foundation-model alignment, unsupervised pre-training, imitation and reinforcement learning, and scale large-model training pipelines. Collaborate cross-functionally, mentor engineers, conduct research, and productionize foundation models for safe, reliable real-world driving.
Top Skills:
SparkFoundation ModelsImitation LearningNumpyPandasPythonPyTorchReinforcement LearningUnsupervised Pre-Training
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
