Online Inference Serving
​OpenAI GPT-2 Text Generation (TF 1) Inception Image Classifier (TF 2) BASNet Object Detector (Custom Public Predictor) FastAI Sentiment Analyzer (Custom PyTorch Predictor) aitextgen GPT-2 Text Generation (Custom PyTorch Predictor)
The CoreWeave Kubernetes Cloud allows for easy serving of machine learning models. The models can be sourced from a range of storage backends, including Amazon S3 and a CoreWeave ReadWriteMany Persistent Volume. After deployment the inference engine auto scales the containers based on demand to swiftly fulfill user requests and scales down as load decreases to not waste GPU resources. Allocating new resources and scaling up a container usually takes 30 seconds, allowing a significantly more responsive service than depending on scaling of hypervisor backed instances of other cloud providers.
Well supported Open Source tools back the inference stack. Knative Serving acts as the serverless runtime, managing auto scaling, revision control and canary deployments. Kubeflow Serving provides an easy to use interface, via Kubernetes resource definitions to deploy models without having to worry about correctly configuring the underlying framework (ie. Tensorflow Serving). The examples in this repository are tailored specifically for common use cases, and there are many more examples in the KFServing repostiory that can be used directly in your namespace.
Autoscaling is enabled by default for all Inference Services. The autoscaling parameters have been pre-configured for GPU based workloads, where a large dataset usually needs to be loaded into GPU VRAM before serving can begin.
KNative Parameter | Value | Description |
tick-interval | 15s | Auto scaling decision are made every 10 seconds |
stable-window | 180s | The time period average concurrency is measured over in stable mode |
container-concurrency-target-percentage | 90% | Scale to load all containers at 100% of their configured concurrency |
max-scale-up-rate | 1.5 | Scale up at a maximum of 50% of current capacity or 1 container (whichever is larger) per 10 seconds |
max-scale-down-rate | 1.01 | Scale down at a maximum of 1% of current capacity or 1 container (whichever is larger) per 10 seconds |
scale-to-zero-grace-period | 30m | If no requests have been received for 30 minutes, a service will be scaled to zero and not use any resources. This behavior can be disabled by setting minReplicas to 1 in the service spec |
If concurrent requests exceeds the current scaled-for request volume by 200% during a period of 30 seconds, the autoscaler enters "panic mode" and starts scaling containers faster than the normal 120s stable window. Some of these settings, such as stable window can be modified using annotations on the InferenceService. Refer to the KNative documentation for more information.
For on demand customers, billing is done on a per minute basis when containers are running. Scale To Zero allows rarely used models to occur no costs, while still be available to receive requests. Please note that when an InferenceService is scaled to zero due to being idle, it usually takes 30-50 seconds depending on model size until the first API request is answered. If this is unacceptable, we recommend setting minReplicas to 1.
Models can be served directly from Amazon S3, which is practical for testing and multi cloud deployment. For faster container launch times in a production environment it is recommended to cache the model in a ReadWriteMany persistent volume on CoreWeave storage. The model can be written to a PVC from any container, such as SSH or Jupyter.
Due to restrictions in KFServe, all node selectors are not available to be used in the InferenceService definition. The annotation serving.kubeflow.org/gke-accelerator should be set to contain the label of the GPU.
