← Back to Search

Staff Python Software Engineer

Company: neptune.ai
Location: Anywhere in the World
Type: Contract
Posted: Mar 18, 2025
Views: 16
python azure google cloud api pytorch software engineer

We seek an experienced Staff Python Software Engineer to join our fully remote team. In this role, you will be part of the squad responsible for developing and maintaining our Python client libraries-APIs for logging and querying model-building metadata. As a key contributor, you will shape their architecture, optimize performance, and ensure seamless integration with machine learning workflows. This role requires a deep understanding of Python, user-friendly API design, and performance optimization to deliver a robust and scalable experience for our users.

Our tech stack (the bigger the overlap, the better):

  • Languages: Python.
  • Cloud platforms: Microsoft Azure, Google Cloud Platform (GCP).
  • Others: Protobufs, gRPC, Swagger.

Responsibilities:

  • Develop and Maintain the Python Client Library: Contribute to the design, development, and maintenance of our Python API for logging and querying model-building metadata.
  • Code Quality & Reviews: Ensure high-quality, maintainable, well-documented code through active development and rigorous code reviews.
  • Technical Leadership: Drive 1-2 key initiatives, working closely with 3-5 engineers to ensure smooth execution, clear task breakdowns, and effective communication of progress and challenges.
  • Architecture & Performance Optimization: Design scalable and efficient solutions for API interactions, ensuring high performance and reliability in large-scale ML workflows.
  • Innovation & Best Practices: Identify and propose enhancements to the Python client library and its surrounding ecosystem to improve user experience and performance.

You might be a fit if you have:

  • Minimum of 6 years of professional software development experience.
  • Strong proficiency in algorithmic thinking and problem-solving.
  • Passion for performance optimization, both on architecture & component level.
  • Concise and logical communication, both written and verbal.
  • Ability to balance business goals with technical purity, especially when adapting to high-paced, shifting priorities.
  • Experience in user-friendly API design.
  • Experience in leading 3-5 people engineering teams.

We offer:

  • Flexibility: 100% remote work with offices (co-works) in Warsaw/Wrocław/Poznań/Kraków available and flexible working hours;
  • Share in our success: Participate in the Employee Stock Option Plan and be part of our growth journey;
  • Time off: 20 paid service-free days per year;
  • Ownership and impact: Space to take action, bring your ideas to life, and make a real impact.

About the Company

Name: neptune.ai
Location: USA/Poland

Experiment tracker purpose-built for foundation model training

Monitor thousands of per-layer metrics—losses, gradients, and activations—at any scale. Visualize them with no lag and no missed spikes. Drill down into logs and debug training issues fast. Keep your model training stable while reducing wasted GPU cycles.

We want to give AI researchers the same level of control and confidence when training models as software developers have when shipping applications.

We are a growing team of 50+ people, representing a few nationalities spread across multiple countries – Poland mainly, but also Armenia, Egypt, France, Portugal, Spain, Switzerland, UK, and more. Check out our About us page to e-meet us all!