google / google/saxml

Python server performance

Open
#1 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
156
Forks
35
PR merge metrics
No merged PRs in 30d

Description

Thank you for making saxml open source! I'm one of the many people who think this project is very useful.

I am interested in how well the [Python server](https://github.com/google/saxml/blob/3e6f110bd4a541d47d8a066eebe1aa7ad3389cd4/saxml/server/model_service_main.py#L148-L156) works. I know that GPU kernels can be launched well with Python to make good use of GPUs. Could Python listen to network requests and handle them efficiently?

If not, should this Python server be used as a prototype for a C++ server that loads and runs AOT-compiled JAX programs (https://jax.readthedocs.io/en/latest/aot.html)? Thanks!

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.