All Projects → elixir-nx → Nx

elixir-nx / Nx

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

Programming Languages

elixir
2628 projects

Projects that are alternatives of or similar to Nx

Compute.scala
Scientific computing with N-dimensional arrays
Stars: ✭ 191 (-83.14%)
Mutual labels:  gpu, tensor
Ocaml Torch
OCaml bindings for PyTorch
Stars: ✭ 308 (-72.82%)
Mutual labels:  gpu, tensor
Norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
Stars: ✭ 211 (-81.38%)
Mutual labels:  gpu, tensor
Deepnet
Deep.Net machine learning framework for F#
Stars: ✭ 99 (-91.26%)
Mutual labels:  gpu, tensor
Libxsmm
Library for specialized dense and sparse matrix operations, and deep learning primitives.
Stars: ✭ 518 (-54.28%)
Mutual labels:  jit, tensor
Pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Stars: ✭ 52,811 (+4561.17%)
Mutual labels:  gpu, tensor
Occa
JIT Compilation for Multiple Architectures: C++, OpenMP, CUDA, HIP, OpenCL, Metal
Stars: ✭ 230 (-79.7%)
Mutual labels:  jit, gpu
Tensorflow Gpu Macosx
Unoffcial NVIDIA CUDA GPU support version of Google Tensorflow for MAC OSX
Stars: ✭ 103 (-90.91%)
Mutual labels:  gpu, tensor
Megengine
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
Stars: ✭ 4,081 (+260.19%)
Mutual labels:  gpu, tensor
Ilgpu
ILGPU JIT Compiler for high-performance .Net GPU programs
Stars: ✭ 374 (-66.99%)
Mutual labels:  jit, gpu
Hyperlearn
50% faster, 50% less RAM Machine Learning. Numba rewritten Sklearn. SVD, NNMF, PCA, LinearReg, RidgeReg, Randomized, Truncated SVD/PCA, CSR Matrices all 50+% faster
Stars: ✭ 1,204 (+6.27%)
Mutual labels:  gpu, tensor
Tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Stars: ✭ 7,494 (+561.43%)
Mutual labels:  gpu, tensor
Laser
The HPC toolbox: fused matrix multiplication, convolution, data-parallel strided tensor primitives, OpenMP facilities, SIMD, JIT Assembler, CPU detection, state-of-the-art vectorized BLAS for floats and integers
Stars: ✭ 191 (-83.14%)
Mutual labels:  jit, tensor
Diffsharp
DiffSharp: Differentiable Functional Programming
Stars: ✭ 365 (-67.78%)
Mutual labels:  gpu, tensor
Cupy
NumPy & SciPy for GPU
Stars: ✭ 5,625 (+396.47%)
Mutual labels:  gpu, tensor
Drlkit
A High Level Python Deep Reinforcement Learning library. Great for beginners, prototyping and quickly comparing algorithms
Stars: ✭ 29 (-97.44%)
Mutual labels:  gpu, tensor
Gpuview
A lightweight web dashboard for monitoring GPU usage
Stars: ✭ 57 (-94.97%)
Mutual labels:  gpu
Aiopen
AIOpen是一个按人工智能三要素(数据、算法、算力)进行AI开源项目分类的汇集项目,项目致力于跟踪目前人工智能(AI)的深度学习(DL)开源项目,并尽可能地罗列目前的开源项目,同时加入了一些曾经研究过的代码。通过这些开源项目,使初次接触AI的人们对人工智能(深度学习)有更清晰和更全面的了解。
Stars: ✭ 62 (-94.53%)
Mutual labels:  gpu
Heteroflow
Concurrent CPU-GPU Programming using Task Models
Stars: ✭ 57 (-94.97%)
Mutual labels:  gpu
Ncnn Android Styletransfer
The style transfer android example
Stars: ✭ 54 (-95.23%)
Mutual labels:  gpu

This repository currently holds the following projects:

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

  • EXLA - Elixir client for Google's XLA with Nx compiler integration and JIT/AOT compilation to the CPU/GPU

  • Torchx - Elixir client to LibTorch with Nx backend integration

Each has their own README, which you can access above to learn more. They will be extracted to their own repository in the future. Examples and benchmarks are available in the EXLA project.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].