tensor · GitHub Topics | GitHub 中文社区

pytorch

@pytorch • Meta

#计算机科学# PyTorch 是一个开源的Python机器学习库，基于Torch，底层由C++实现。

neural-network autograd gpu numpy deep-learning tensor python machine-learning

Python84.37 k

32 分钟前

tvm

Apache软件基金会@apache

#计算机科学# Open deep learning compiler stack for cpu, gpu and specialized accelerators

翻译 - 针对cpu，gpu和专用加速器的开放式深度学习编译器堆栈

compiler tensor deep-learning gpu opencl metal performance javascript rocm tvm

Python11.81 k

1 天前

burn

@tracel-ai

#计算机科学# burn 是一个Rust的深度学习框架

autodiff deep-learning machine-learning rust scientific-computing ndarray tensor neural-network pytorch autotune

Rust8.98 k

4 天前

einops

Alex Rogozhnikov@arogozhnikov

#计算机科学# Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

翻译 - 重塑了深度学习操作（支持tf，pytorch，chainer，gluon等）

deep-learning pytorch tensorflow numpy gluon cupy chainer keras tensor jax

Python8.53 k

3 天前

MegEngine

@MegEngine

#计算机科学# MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

deep-learning machine-learning megengine tensor autograd python gpu numpy

C++4.76 k

1 个月前

@elixir-nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

翻译 - 用于CPU / GPU / TPU的数值Elixir和Elixir XLA绑定

gpu numerical elixir tensor

Elixir2.66 k

6 天前

executorch

@pytorch • Meta

#计算机科学# On-device AI across mobile, embedded and edge for PyTorch

deep-learning embedded machine-learning mobile neural-network tensor gpu

C++2.21 k

3 小时前

llama2.mojo

@tairov

Inference Llama 2 in one file of pure 🔥

inference llama llama2 modular mojo performance simd vectorization parallelize tensor

Mojo2.1 k

6 个月前

koila

@rentruewang

#计算机科学# Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

翻译 - 仅用 1 行代码即可防止 PyTorch 的“CUDA 错误：内存不足”。

pytorch torch lazy-evaluation out-of-memory python numpy machine-learning deep-learning tensor memory-management

Python1.82 k

1 个月前

NiuTensor

@NiuTrans

#计算机科学# NiuTensor is an open-source toolkit developed by a joint team from NLP Lab. at Northeastern University and the NiuTrans Team. It provides tensor utilities to create and train neural networks.

翻译 - NiuTensor是由NLP Lab的联合团队开发的开源工具包。在东北大学和NiuTrans团队。它提供了张量实用程序来创建和训练神经网络。

deep-learning artificial-neural-networks tensor machine-learning

C++379

3 年前

PuzzleLib

@puzzlelib

#计算机科学# Deep Learning framework with NVIDIA & AMD support

翻译 - 具有NVIDIA和AMD支持的深度学习框架

deep-learning data-science neural-network machine-learning library python deep-neural-networks gpu tensor numpy

Python31

5 年前