Skip to main content

A python SDK for submarine

Project description


PySubmarine is aiming to ease the ML engineer's life by providing a set of libraries.

It includes a high-level out-of-box ML library like deepFM, FM, etc. low-level library to interact with submarine like creating experiment, tracking experiment metrics, parameters.

Package setup

  • Install latest version of pysubmarine
git clone
cd submarine/submarine-sdk/pysubmarine
pip install .

# install extras_require
## bash
pip install -e .[tf2,pytorch]
## zsh
pip install -e ".[tf2,pytorch]"
  • Install package from pypi
pip install apache-submarine

Easy-to-use model trainers

Submarine experiment management

Makes it easy to run distributed or non-distributed TensorFlow, PyTorch experiments on Kubernetes.


See Python Development in the documentation subproject.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

apache_submarine-0.8.0-py3-none-any.whl (212.1 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page