dfdx

Deep learning library

A deep learning library for Rust with GPU acceleration and ergonomic API.

Deep learning in Rust, with shape checked tensors and neural networks

GitHub

2k stars
35 watching
99 forks
Language: Rust
last commit: 6 months ago
Linked from 2 awesome lists

autodiffautodifferentiationautogradbackpropagationcudacuda-kernelscuda-supportcuda-toolkitcudnndeep-learningdeep-neural-networksgpugpu-accelerationgpu-computingmachine-learningneural-networkrustrust-langtensor

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
nicholas-leonard/dp A deep learning library for streamlining research and development using the Torch7 distribution. 343
dmarnerides/pydlt A PyTorch-based toolbox for building and training deep learning models with ease. 204
yixuan/minidnn A C++ library implementing deep neural networks with good performance and modularity 399
nitishsrivastava/deepnet A collection of GPU-accelerated deep learning algorithms implemented in Python 895
mosdeo/lkydeepnn A header-only C++ library for training and deploying deep neural networks on embedded systems 49
uncomplicate/deep-diamond A fast Clojure library for computing with tensors and neural networks on CPUs and GPUs. 433
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,093
amd/opencl-caffe An OpenCL implementation of Caffe, a mainstream DNN framework. 518
vict0rsch/deep_learning A collection of tutorials and resources on implementing deep learning models using Python libraries such as Keras and Lasagne. 426
hannes-brt/hebel A Python library for building and training neural networks using GPU acceleration 1,169
dusty-nv/ros_deep_learning A set of ROS/ROS2 nodes for deep learning inference on NVIDIA devices 907
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 337
rksltnl/deep-metric-learning-cvpr16 A software framework for building deep metric learning models using lifted structured feature embedding 346
dustinstansbury/medal A Matlab environment for training and using various deep learning architectures 109
kuz/deepmind-atari-deep-q-learner An implementation of a deep reinforcement learning architecture for playing Atari games 1,828