Senior Software Engineer, Machine Learning

At Treasure Data, we’re on a mission to radically simplify how companies use data to create connected customer experiences. Our sophisticated cloud-based customer data platform drives operational efficiency across the enterprise to deliver powerful business outcomes in a way that’s safe, flexible, and secure. 

We are thrilled that Gartner Magic Quadrant has recognized Treasure Data as a Leader in Customer Data Platforms for 2024! It's an honor to be acknowledged for our efforts in advancing the CDP industry with cutting-edge AI and real-time capabilities.

Furthermore, Treasure Data employees are enthusiastic, data-driven, and customer-obsessed. We are a team of drivers—self-starters who take initiative, anticipate needs, and proactively jump in to solve problems. Our actions reflect our values of honesty, reliability, openness, and humility. 

About the Role:

The Machine Learning (ML) team is dedicated to empowering our customers to make data-driven decisions using insights derived from ML solutions. We are revamping our ML infrastructure to make it more scalable and maintainable, and continue to build ML solutions on top of it to be integrated with our award-winning CDP. By integrating these solutions we aim to reduce the hurdles and enable individuals with limited data science experience, such as marketers, to harness the power of AI.

As an example of the strategic ML investments, our PrecisionML solutions, including Recency, Frequency, Monetary (RFM) and Next Best Product (NBP), have been rolled out, with more ML solutions in the pipeline. The team’s goal is to empower every customer to take advantage of AI to drive business value without the need for extensive knowledge of data science. We are committed to driving this multi-year initiative of democratizing AI for every customer.

We are seeking strong ML engineers to join the ML team and help us sharpen the ML vision and deliver more solutions to satisfy our customers’ needs. You will be working closely with other ML engineers and people from cross-functional teams.

You will be a good fit if you:

• Are a self-driven, organized, and independent individual who proactively takes initiatives, anticipates needs, and solves problems to contribute to delivering values to our customers.
• Have a strong sense of ownership and responsibility to get things done.
• Have a growth mindset; are curious to learn new things and adaptive to changes.
• Know how to navigate ambiguity and thrive in uncertain environments, consistently driving work forward.
• Are passionate about productizing ML products, knowing how to make practical trade-offs when turning ideas into working software.
• Excel in adapting communication styles and simplifying complex technical concepts for diverse audiences, ensuring clear and effective communication.
• Enjoy working in a collaborative work environment with people from diverse backgrounds.

This is an ideal position for those with not only data science and machine learning skills, but also cloud service engineering skills for developing, deploying, and operating these critical ML products.

Responsibilities & Duties:

• Design and develop ML products to be integrated with our CDP.
• Own specific technical areas, drive and execute ML product projects, track progress and mitigate risks by collaborating closely with product managers, UX designers, architects, engineers, and stakeholders from other cross-functional teams.
• Contribute to defining system architecture for the ML products and implementing specific components to enhance the user experience.
• Design and implement performant, scalable EtLT data pipelines, considering an ML model’s lifecycle (training and inference).
• Take responsibility for technical problem solving and meeting ML product objectives creatively in ambiguous scenarios.
• Participate in the on-call rotation for production support.
• Drive best practices including ML research methodologies, coding standards, code reviews, source control management, development processes, build processes, testing and release, and operational excellence.

Required Qualifications: 

• Advanced degree in computer science, data science, machine learning, or related field, or equivalent work experience.
• 6+ years of professional experience in software engineering designing and building ML-driven products, with at least 3 years focused on production ML systems.
• Fundamental knowledge of Data Engineering and extensive experience in developing and deploying ML models, as well as building and maintaining ML pipelines and products.
• Experience in applying scientific method: hypothesis formulation and testing, exploratory data analysis, cross-validation, reproducible research, structured reporting/documentation like in IMRAD format, result explanation and presentation.
• Experience designing, deploying, and operating scalable ML systems in production. This includes responsibility for model selection, performance benchmarking, and lifecycle management to solve real-world business problems.
• Proficiency in Python and general ML ecosystem tooling in data processing and modelling (such as NumPy, pandas, scikit-learn, PyTorch, etc.).
• Experience in designing and building products using public cloud services such as AWS.
• Excellent verbal and written communication skills in English, and ability to convey research findings and implications to both technical and non-technical audiences.
• Ability to work effectively in cross-functional and distributed teams across different time

Nice to have: 

• Experienced with security design principles and best practices.
• Experience working with big data technologies such as Hive, Trino, Spark, BigQuery, and Redshift.
• OSS contribution experiences.

Physical Requirments:

• We are open to hiring candidates to work in Tokyo, Japan. Relocation is not supported.
• Per current company’s hybrid working policy the candidate is required to work in the Tokyo office 3 days a week.

About Treasure Data:

Treasure Data is the Intelligent Customer Data Platform (CDP) built for enterprise scale and powered by AI. Recognized as a Leader by Forrester and IDC, Treasure Data empowers the world’s largest and most innovative companies to deliver hyper-personalized customer experiences at scale that increase revenue, reduce costs, and build trust.

Through unique capabilities such as the Diamond Record, AI Agent Foundry, and AI Decisioning with Real-Time Personalization, Treasure Data enables marketing and CX teams to personalize cross-channel engagement in real-time, optimize marketing spend while increasing ROI, and drive customer lifetime value through more intelligent retention and loyalty.

Our Dedication to You:

We value and promote diversity, equity, inclusion, and belonging in all aspects of our business and at all levels. Success comes from acknowledging, welcoming, and incorporating diverse perspectives. 

Diverse representation alone is not the desired outcome. We also strive to create an inclusive culture that encourages growth, ownership of your role, and achieving innovation in new and unique ways. Your voice will be heard, and we will help amplify it.