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: about 2 years 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:

RepositoryDescriptionStars
nicholas-leonard/dpA deep learning library for streamlining research and development using the Torch7 distribution.343
dmarnerides/pydltA PyTorch-based toolbox for building and training deep learning models with ease.204
yixuan/minidnnA C++ library implementing deep neural networks with good performance and modularity399
nitishsrivastava/deepnetA collection of GPU-accelerated deep learning algorithms implemented in Python895
mosdeo/lkydeepnnA header-only C++ library for training and deploying deep neural networks on embedded systems49
uncomplicate/deep-diamondA fast Clojure library for computing with tensors and neural networks on CPUs and GPUs.433
eduardoleao052/js-pytorchA JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers.1,093
amd/opencl-caffeAn OpenCL implementation of Caffe, a mainstream DNN framework.518
vict0rsch/deep_learningA collection of tutorials and resources on implementing deep learning models using Python libraries such as Keras and Lasagne.426
hannes-brt/hebelA Python library for building and training neural networks using GPU acceleration1,169
dusty-nv/ros_deep_learningA set of ROS/ROS2 nodes for deep learning inference on NVIDIA devices907
torch/cutorchProvides a CUDA backend for the PyTorch deep learning framework337
rksltnl/deep-metric-learning-cvpr16A software framework for building deep metric learning models using lifted structured feature embedding346
dustinstansbury/medalA Matlab environment for training and using various deep learning architectures109
kuz/deepmind-atari-deep-q-learnerAn implementation of a deep reinforcement learning architecture for playing Atari games1,828