Open Source Alternatives to PyTorch
What is PyTorch?
PyTorch is an open-source machine learning framework that provides tensor computation with strong GPU acceleration and deep neural networks built on a tape-based autograd system.. There are currently 16 open-source alternatives to PyTorch, with a combined total of 54.6k GitHub stars. The most common language among these projects is Rust.
Showing 16 open-source alternatives
A minimalist, high-performance machine learning framework for Rust with a focus on serverless inference and GPU support.
A Rust-based deep learning framework and tensor library optimized for flexibility, efficiency, and cross-platform portability.
A library for building and evaluating mathematical expressions and neural networks in Go, with automatic differentiation and GPU support.
Multi-dimensional arrays (tensors) and numerical definitions for Elixir, enabling machine learning and scientific computing.
A Julia machine learning framework providing a unified interface and meta-algorithms for over 200 models.
A neural network library for Elixir built on Nx, providing functional, model creation, and training APIs for deep learning.
An accelerated machine learning framework for Go, offering a PyTorch/Jax/TensorFlow-like experience with support for CPUs, GPUs, TPUs, and WASM.
A deep learning framework for Julia with GPU support and automatic differentiation using dynamic computational graphs.
A deep learning JavaScript library built from scratch with PyTorch-like syntax and GPU acceleration via GPU.js.
A collection of tutorials and resources to help developers learn JAX, Flax, and Haiku for machine learning.
A Clojure library for dynamic neural network graphs with pluggable tensor backends, inspired by PyTorch.
A core scientific computing library for Crystal providing n-dimensional tensors, linear algebra, GPU acceleration, and automatic differentiation.
A pure, immutable module system for JAX that replaces PyTorch-style imperative coding with declarative parameter trees.
A fast, flexible, and compact deep learning framework for Julia that runs on CPU and CUDA GPU.
A lightweight, platform-independent tensor library with autograd for the JVM, accelerated by OpenCL.
Type-safe n-dimensional arrays (tensors) in Scala 3 with compile-time shape, axis label, and data type validation.
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.