torsh-python 0.1.2

Python bindings for ToRSh - PyTorch-compatible deep learning in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

"""
ToRSh Python Bindings

PyTorch-compatible deep learning framework implemented in Rust.

This package provides Python bindings for the ToRSh deep learning framework,
offering high-performance tensor operations and neural network functionality.
"""

from .rstorch import *  # noqa: F401, F403

__all__ = [
    # Version
    "__version__",
    # Main classes
    "PyDevice",
    "PyDType",
    "TorshError",
    # Device constants
    "cpu",
    # Device utility functions
    "device_count",
    "is_available",
    "cuda_is_available",
    "mps_is_available",
    "get_device_name",
    # DType constants
    "float32",
    "f32",
    "float64",
    "f64",
    "int8",
    "i8",
    "int16",
    "i16",
    "int32",
    "i32",
    "int64",
    "i64",
    "uint8",
    "u8",
    "uint32",
    "u32",
    "uint64",
    "u64",
    "bool",
    # PyTorch-style aliases
    "float",
    "double",
    "long",
    "int",
    "short",
    "byte",
    # DType utility functions
    "promote_types",
    "result_type",
    "can_operate",
]