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.345.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.345-py3-none-any.whl (838.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: workbench-0.8.345.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.345.tar.gz
Algorithm Hash digest
SHA256 388ca28fe7301a96d2a2678a481cadeda7da86ddc2512b23ba1c76ed14189448
MD5 12609b24579da20afbe13c68a6deac1c
BLAKE2b-256 1314e5df7064e74f8d21d8bb59d3d41616347f626cda34d2e1af22f5eca2b818

See more details on using hashes here.

Provenance

The following attestation bundles were made for workbench-0.8.345.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.345-py3-none-any.whl.

File metadata

  • Download URL: workbench-0.8.345-py3-none-any.whl
  • Upload date:
  • Size: 838.1 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.345-py3-none-any.whl
Algorithm Hash digest
SHA256 a81482a77b0d39deceb2b04dda895be4c3480f3153fa71a97324a355e08dcfd1
MD5 7e182af65142e64bc2a54deb5d081515
BLAKE2b-256 ea820778b1e52656f631308b5fd573756bfc17e949f0abaafba9786578588cf2

See more details on using hashes here.

Provenance

The following attestation bundles were made for workbench-0.8.345-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