Skip to main content

Workbench: A Dashboard and Python API for creating and deploying AWS SageMaker Model Pipelines

Project description

Live Dashboard Demo

You can explore a live demo of the Workbench Dashboard at: Workbench Dashboard Demo

Recent News

Chemprop Models! All the rage for the Open ADMET Challenge.

ADMET Workbench now supports:

  • Single Task Chemprop Models
  • Multi Task Chemprop Models
  • Chemprop Hybrid Models (MPNN + Descriptors)
  • Foundation Chemprop Models (CheMeleon Pretrained)

Examples:

References

Chemprop Action Shots!

theme_dark
theme_quartz

Welcome to ADMET Workbench

The ADMET Workbench framework makes AWS® both easier to use and more powerful. Workbench handles all the details around updating and managing a complex set of AWS Services. With a simple-to-use Python API and a beautiful set of web interfaces, Workbench makes creating AWS ML pipelines a snap. It also dramatically improves both the usability and visibility across the entire spectrum of services: Glue Job, Athena, Feature Store, Models, and Endpoints, Workbench makes it easy to build production ready, AWS powered, machine learning pipelines.

workbench_new_light

Full AWS ML OverView

  • Health Monitoring 🟢
  • Dynamic Updates
  • High Level Summary

Drill-Down Views

  • Incoming Data
  • Glue Jobs
  • DataSources
  • FeatureSets
  • Models
  • Endpoints

Private SaaS Architecture

Secure your Data, Empower your ML Pipelines

ADMET Workbench is architected as a Private SaaS (also called BYOC: Bring Your Own Cloud). This hybrid architecture is the ultimate solution for businesses that prioritize data control and security. Workbench deploys as an AWS Stack within your own cloud environment, ensuring compliance with stringent corporate and regulatory standards. It offers the flexibility to tailor solutions to your specific business needs through our comprehensive plugin support. By using Workbench, you maintain absolute control over your data while benefiting from the power, security, and scalability of AWS cloud services. Workbench Private SaaS Architecture

private_saas_compare

API Installation

For typical use (the API, REPL, dashboard, training pipelines):

  • pip install 'workbench[all]' Full install — recommended
  • workbench Runs the Workbench REPL / initial setup

pip install workbench (no extras) is intentionally lightweight — it's the endpoint-safe surface that ships inside SageMaker inference containers (and the lambdas / scripts that just need to invoke endpoints). See Installation extras below for the breakdown.

For the full instructions for connecting your AWS Account see:

ADMET Workbench up on the AWS Marketplace

Powered by AWS® to accelerate your Machine Learning Pipelines development with our new Dashboard for ML Pipelines. Getting started with Workbench is a snap and can be billed through AWS.

ADMET Workbench Presentations

Even though ADMET Workbench makes AWS easier, it's taking something very complex (the full set of AWS ML Pipelines/Services) and making it less complex. Workbench has a depth and breadth of functionality so we've provided higher level conceptual documentation See: Workbench Presentations

workbench_api

ADMET Workbench Documentation

The ADMET Workbench documentation Workbench Docs covers the Python API in depth and contains code examples. The documentation is fully searchable and fairly comprehensive.

The code examples are provided in the Github repo examples/ directory. For a full code listing of any example please visit our Workbench Examples

Questions?

The SuperCowPowers team is happy to answer any questions you may have about AWS and Workbench. Please contact us at workbench@supercowpowers.com or chat us up on Discord

ADMET Workbench Beta Program

Using ADMET Workbench will minimize the time and manpower needed to incorporate AWS ML into your organization. If your company would like to be a Workbench Beta Tester, contact us at workbench@supercowpowers.com.

Installation extras

Workbench's dependencies are organized so you can install exactly what you need. The workbench.endpoints.* surface is enforced by a CI smoke test that runs the lightweight install in a clean venv and verifies every module under that namespace imports without any extras pulled in — which is what makes the base install safe to drop into a SageMaker endpoint container or a lambda.

pip install workbench               # Endpoint-safe core only:
                                    #   boto3, awswrangler, numpy, pandas,
                                    #   sklearn, scipy, rdkit, joblib
                                    # Use in lambdas, endpoint containers,
                                    # or anywhere you just need to invoke
                                    # endpoints and read/write S3.

pip install 'workbench[aws]'        # + sagemaker SDK + aiobotocore + redis +
                                    #   cryptography. Needed for the orchestration
                                    #   side: building pipelines, deploying
                                    #   endpoints, talking to SageMaker training.

pip install 'workbench[modeling]'   # + xgboost, umap-learn, mordred,
                                    #   cleanlab, ipython. Training-time ML
                                    #   libs (SageMaker training containers
                                    #   have most of these pre-installed).

pip install 'workbench[ui]'         # + plotly, dash, dash-ag-grid,
                                    #   matplotlib. The Workbench Dashboard.

pip install 'workbench[dev]'        # + pytest, pytest-xdist, coverage,
                                    #   flake8, black. Local development.

pip install 'workbench[all]'        # All of the above — typical full install
                                    #   for interactive use, dashboards, and
                                    #   building/deploying pipelines.

Note: shells may interpret square brackets as globs, so the quotes are needed.

Model-script code running inside SageMaker endpoint containers should import exclusively from workbench.endpoints.* — that's the contract the endpoint-import-smoke CI job enforces. See workbench/endpoints/__init__.py for the full surface.

Contributions

If you'd like to contribute to the ADMET Workbench project, you're more than welcome. All contributions will fall under the existing project license. If you are interested in contributing or have questions please feel free to contact us at workbench@supercowpowers.com.

® Amazon Web Services, AWS, the Powered by AWS logo, are trademarks of Amazon.com, Inc. or its affiliates

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

workbench-0.8.339.tar.gz (3.1 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

workbench-0.8.339-py3-none-any.whl (819.3 kB view details)

Uploaded Python 3

File details

Details for the file workbench-0.8.339.tar.gz.

File metadata

  • Download URL: workbench-0.8.339.tar.gz
  • Upload date:
  • Size: 3.1 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for workbench-0.8.339.tar.gz
Algorithm Hash digest
SHA256 c2892829bf417211dfb5875e8720ea8070ba95615548ec3f339a9a90825c164f
MD5 aa9c62964a001f01991f287a3af9568b
BLAKE2b-256 c04b97fa0edd92bfb61beac45224dab5bf329e7a1d50f3485a45929a5d8ecf9a

See more details on using hashes here.

Provenance

The following attestation bundles were made for workbench-0.8.339.tar.gz:

Publisher: publish.yml on SuperCowPowers/workbench

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file workbench-0.8.339-py3-none-any.whl.

File metadata

  • Download URL: workbench-0.8.339-py3-none-any.whl
  • Upload date:
  • Size: 819.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for workbench-0.8.339-py3-none-any.whl
Algorithm Hash digest
SHA256 9be9a6fa63a5f841af4f412a56f519e568e148d142d75b0296e3db408f4825f7
MD5 c0f4405875d06ea227d4d1869d4544c8
BLAKE2b-256 f6ebb837c0c0c98201082ac51c43aa4e3fb333a31a5c979f544c4a6a6a78dd22

See more details on using hashes here.

Provenance

The following attestation bundles were made for workbench-0.8.339-py3-none-any.whl:

Publisher: publish.yml on SuperCowPowers/workbench

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

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