alphai 0.0.7__py3-none-any.whl → 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- alphai/__init__.py +8 -4
- alphai/auth.py +362 -0
- alphai/cli.py +1015 -0
- alphai/client.py +400 -0
- alphai/config.py +88 -0
- alphai/docker.py +764 -0
- alphai/utils.py +192 -0
- alphai-0.1.0.dist-info/METADATA +394 -0
- alphai-0.1.0.dist-info/RECORD +12 -0
- {alphai-0.0.7.dist-info → alphai-0.1.0.dist-info}/WHEEL +2 -1
- alphai-0.1.0.dist-info/entry_points.txt +2 -0
- alphai-0.1.0.dist-info/top_level.txt +1 -0
- alphai/alphai.py +0 -786
- alphai/api/client.py +0 -0
- alphai/benchmarking/benchmarker.py +0 -37
- alphai/client/__init__.py +0 -0
- alphai/client/client.py +0 -382
- alphai/profilers/__init__.py +0 -0
- alphai/profilers/configs_base.py +0 -7
- alphai/profilers/jax.py +0 -37
- alphai/profilers/pytorch.py +0 -83
- alphai/profilers/pytorch_utils.py +0 -419
- alphai/util.py +0 -19
- alphai-0.0.7.dist-info/LICENSE +0 -201
- alphai-0.0.7.dist-info/METADATA +0 -125
- alphai-0.0.7.dist-info/RECORD +0 -16
alphai-0.0.7.dist-info/METADATA
DELETED
|
@@ -1,125 +0,0 @@
|
|
|
1
|
-
Metadata-Version: 2.1
|
|
2
|
-
Name: alphai
|
|
3
|
-
Version: 0.0.7
|
|
4
|
-
Summary:
|
|
5
|
-
Author: Andrew Chang
|
|
6
|
-
Requires-Python: >=3.9,<3.12
|
|
7
|
-
Classifier: Programming Language :: Python :: 3
|
|
8
|
-
Classifier: Programming Language :: Python :: 3.9
|
|
9
|
-
Classifier: Programming Language :: Python :: 3.10
|
|
10
|
-
Classifier: Programming Language :: Python :: 3.11
|
|
11
|
-
Provides-Extra: docs
|
|
12
|
-
Provides-Extra: jupyter
|
|
13
|
-
Provides-Extra: torch
|
|
14
|
-
Requires-Dist: datasets (>=2.16.1,<3.0.0)
|
|
15
|
-
Requires-Dist: gradio (>=4.14.0,<5.0.0)
|
|
16
|
-
Requires-Dist: holistictraceanalysis (>=0.2.0,<0.3.0)
|
|
17
|
-
Requires-Dist: jh-client (>=1.3.0,<2.0.0)
|
|
18
|
-
Requires-Dist: jupyter (>=1.0.0,<2.0.0) ; extra == "jupyter"
|
|
19
|
-
Requires-Dist: jupyterlab (>=4.0.9,<5.0.0) ; extra == "jupyter"
|
|
20
|
-
Requires-Dist: matplotlib (>=3.8.2,<4.0.0) ; extra == "jupyter"
|
|
21
|
-
Requires-Dist: mkdocs (>=1.5.3,<2.0.0) ; extra == "docs"
|
|
22
|
-
Requires-Dist: mkdocs-material (>=9.4.12,<10.0.0) ; extra == "docs"
|
|
23
|
-
Requires-Dist: mkdocs-minify-plugin (>=0.7.2,<0.8.0) ; extra == "docs"
|
|
24
|
-
Requires-Dist: mkdocstrings (>=0.18.0,<0.19.0) ; extra == "docs"
|
|
25
|
-
Requires-Dist: nbserv-client (>=1.4.0,<2.0.0)
|
|
26
|
-
Requires-Dist: openai (>=1.3.5,<2.0.0)
|
|
27
|
-
Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
|
|
28
|
-
Requires-Dist: torch (==2.*) ; extra == "torch"
|
|
29
|
-
Requires-Dist: torchaudio (==2.*) ; extra == "torch"
|
|
30
|
-
Requires-Dist: torchvision (==0.*) ; extra == "torch"
|
|
31
|
-
Requires-Dist: transformers (>=4.35.2,<5.0.0)
|
|
32
|
-
Requires-Dist: wget (>=3.2,<4.0)
|
|
33
|
-
Description-Content-Type: text/markdown
|
|
34
|
-
|
|
35
|
-
# AlphAI
|
|
36
|
-
|
|
37
|
-
AlphAI is a high-level open-source Python toolkit designed for efficient AI development and in-depth GPU profiling. Supporting popular tensor libraries like [PyTorch](https://pytorch.org/get-started/locally/) and [Jax](https://github.com/google/jax), it optimizes developer operations on GPU servers and integrates seamlessly with [American Data Science Labs](https://dashboard.amdatascience.com), offering robust control over remote Jupyter Lab servers and environment runtimes.
|
|
38
|
-
|
|
39
|
-
## Features
|
|
40
|
-
|
|
41
|
-
- **GPU Profiling and Analytics**: Advanced GPU profiling capabilities to maximize resource efficiency and performance.
|
|
42
|
-
- **Benchmarking Tools**: Pythonic, easy-to-use tools for evaluating and comparing model performance.
|
|
43
|
-
- **Remote Jupyter Lab Integration**: Programmatic management of remote Jupyter Lab servers for enhanced productivity.
|
|
44
|
-
- **Local Tensor Model Support**: Streamlines the integration and management of tensor models from providers like Hugging Face.
|
|
45
|
-
- **Tensor Engine Compatibility**: Fully compatible with PyTorch, with upcoming support for Jax and TensorFlow.
|
|
46
|
-
|
|
47
|
-
## Quick Start
|
|
48
|
-
|
|
49
|
-
### Installation
|
|
50
|
-
|
|
51
|
-
Install AlphAI easily using pip:
|
|
52
|
-
|
|
53
|
-
```bash
|
|
54
|
-
pip install alphai
|
|
55
|
-
|
|
56
|
-
# If you'd like to install torch in a Linux machine with CUDA-drivers
|
|
57
|
-
pip install alphai[torch]
|
|
58
|
-
```
|
|
59
|
-
|
|
60
|
-
### Authentication Pre-requisites
|
|
61
|
-
|
|
62
|
-
Although not strictly required to use the computational functions of the alphai package, it is recommended to create an account at [American Data Science](https://dashboard.amdatascience.com) and generate an API key to make use of your two free remote Jupyter Lab servers.
|
|
63
|
-
|
|
64
|
-
You don't need an API key to use the GPU profiling, benchmarking, and generate modules.
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
### Basic Usage
|
|
68
|
-
|
|
69
|
-
Here's a quick example to get started with AlphAI:
|
|
70
|
-
|
|
71
|
-
```python
|
|
72
|
-
from alphai import AlphAI
|
|
73
|
-
|
|
74
|
-
# Initialize AlphAI
|
|
75
|
-
|
|
76
|
-
aai = AlphAI(
|
|
77
|
-
api_key=os.environ.get("ALPHAI_API_KEY"),
|
|
78
|
-
)
|
|
79
|
-
|
|
80
|
-
# Start remote Jupyter Lab servers
|
|
81
|
-
aai.start_server()
|
|
82
|
-
|
|
83
|
-
# Upload to your server's file system
|
|
84
|
-
aai.upload("./main.py")
|
|
85
|
-
|
|
86
|
-
# Start python kernel and run code remotely
|
|
87
|
-
code = "print('Hello world!')"
|
|
88
|
-
aai.run_code(code)
|
|
89
|
-
|
|
90
|
-
```
|
|
91
|
-
|
|
92
|
-
## Documentation and Detailed Usage
|
|
93
|
-
|
|
94
|
-
For more documentation and detailed instructions on how to use AlphAI's various features, please refer to our [Documentation](https://alphai.amdatascience.com).
|
|
95
|
-
|
|
96
|
-
### [Working with Tensor Models](https://alphai.amdatascience.com/americandatascience/alphai/models/hugging-face/)
|
|
97
|
-
|
|
98
|
-
Guidance on integrating and leveraging tensor models.
|
|
99
|
-
|
|
100
|
-
### [GPU Profiling and Analytics](https://alphai.amdatascience.com/americandatascience/alphai/gpu-profiling/)
|
|
101
|
-
|
|
102
|
-
Comprehensive features for GPU profiling and analytics.
|
|
103
|
-
|
|
104
|
-
### [Integration with American Data Science Labs](https://alphai.amdatascience.com/americandatascience/alphai/servers/)
|
|
105
|
-
|
|
106
|
-
Discover the benefits of integrating AlphAI with American Data Science Labs.
|
|
107
|
-
|
|
108
|
-
## System Requirements
|
|
109
|
-
|
|
110
|
-
- Python 3.9+
|
|
111
|
-
- PyTorch (recommnended) or Jax (limited support)
|
|
112
|
-
- Linux OS i.e. Ubuntu 18.04+
|
|
113
|
-
|
|
114
|
-
## Contributing
|
|
115
|
-
|
|
116
|
-
We welcome contributions! Please see our [Contribution Guidelines](https://github.com/americandatascience/alphai/README.md) for more information.
|
|
117
|
-
|
|
118
|
-
## License
|
|
119
|
-
|
|
120
|
-
AlphAI is released under the [Apache 2.0](https://github.com/americandatascience/alphai/LICENSE.txt) license.
|
|
121
|
-
|
|
122
|
-
## Support and Contact
|
|
123
|
-
|
|
124
|
-
For support or inquiries about enterprise solutions, contact us at [info@amdatascience.com](mailto:info@amdatascience.com).
|
|
125
|
-
|
alphai-0.0.7.dist-info/RECORD
DELETED
|
@@ -1,16 +0,0 @@
|
|
|
1
|
-
alphai/__init__.py,sha256=6GT5TD28clMwXd-vad_MxwFl3YbAyf7Xz5eQcLmRKhM,63
|
|
2
|
-
alphai/alphai.py,sha256=c6RxEaCsyYph5RJ4Az4fIDeOKztJKYOFRnO3ACCxGak,30965
|
|
3
|
-
alphai/api/client.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
4
|
-
alphai/benchmarking/benchmarker.py,sha256=n17sgQPxhJK262Qot5SV1foPMl8QInOZWHbg8vfQAFg,1050
|
|
5
|
-
alphai/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
6
|
-
alphai/client/client.py,sha256=VvVWWAhTK4pSnFM1oGmEYywgWQIxUe8Lqo-Q0nCEOK8,12630
|
|
7
|
-
alphai/profilers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
8
|
-
alphai/profilers/configs_base.py,sha256=v8CNk2AI39sWFcJJcyArPk6MWM3UGM9p7uZD9eCggX4,145
|
|
9
|
-
alphai/profilers/jax.py,sha256=dIUsHyfuM6MBrdMMxF1qHnNy6KyYuVtJnHH0EjNzaqs,1165
|
|
10
|
-
alphai/profilers/pytorch.py,sha256=5LFZTXCSInsBPsYZtkGNWwYWI3TQs_d7XngB30Yblw4,2722
|
|
11
|
-
alphai/profilers/pytorch_utils.py,sha256=bIk9qIl8m3XRwXVj1naFAkNIXiflFqU-3t9I7re9QqM,13868
|
|
12
|
-
alphai/util.py,sha256=oryhRFfIS5hpveD_T5N7NIBu8PKDTibfB9z6GmtwFJk,465
|
|
13
|
-
alphai-0.0.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
|
14
|
-
alphai-0.0.7.dist-info/METADATA,sha256=AtRxxpqvbOggGnUYK4ISOA5-93q1pxaen0DwadiAMks,4792
|
|
15
|
-
alphai-0.0.7.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
|
|
16
|
-
alphai-0.0.7.dist-info/RECORD,,
|