tensorizerA tool for fast PyTorch module, model, and tensor serialization + deserialization.tensorizermachinelearningserializationtensorpytorch
adoradA Tensor library in Python, built in C++c++tensordeeplearningmachinegradientscomputervisionimageprocessingopencvmatplotlibaiartificial-intelligenceaugmentationcaercomputer-visioncudadata-sciencedeep-learninggpuimage-classificationimage-processingimage-segmentationmachine-learningneural-networkpythonsegmentationtype-checkingvideo-processing
apache-tvm-cu113Open deep learning compiler stack for cpu, gpu and specialized acceleratorscompilerdeep-learninggpujavascriptmachine-learningmetalopenclperformancerocmspirvtensortvmvulkan
cupy-cuda11xCuPy: NumPy & SciPy for GPUcublascudacudnncupycurandcusolvercusparsecusparseltcutensorgpuncclnumpynvrtcnvtxpythonrocmscipytensor
tb-nightlyTensorBoard lets you watch Tensors Flowtensorflowtensorboardtensormachinelearningvisualizer
cupy-cuda115CuPy: NumPy & SciPy for GPUcublascudacudnncupycurandcusolvercusparsecusparseltcutensorgpuncclnumpynvrtcnvtxpythonrocmscipytensor
tensorboard-data-serverFast data loading for TensorBoardtensorflowtensorboardtensormachinelearningvisualizer
cupy-cuda110CuPy: NumPy & SciPy for GPUcublascudacudnncupycurandcusolvercusparsecusparseltcutensorgpuncclnumpynvrtcnvtxpythonrocmscipytensor
cupy-rocm-5-0CuPy: NumPy & SciPy for GPUcublascudacudnncupycurandcusolvercusparsecusparseltcutensorgpuncclnumpynvrtcnvtxpythonrocmscipytensor
tensorflow-decision-forestsCollection of training and inference decision forest algorithms.tensorflowtensormachinelearningdecisionforestsrandomforestgradientboostedtreesdecision-forestdecision-treesgradient-boostinginterpretabilitykerasmachine-learningmlpythonrandom-forest