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
34 watching
98 forks
Language: Rust
last commit: 4 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 397
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,084
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 GPU-accelerated deep learning 1,169
dusty-nv/ros_deep_learning A set of ROS/ROS2 nodes for deep learning inference on NVIDIA devices 901
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 336
rksltnl/deep-metric-learning-cvpr16 A software framework for building deep metric learning models using lifted structured feature embedding 342
dustinstansbury/medal A Matlab environment for building and training deep neural networks with various architectures. 109
kuz/deepmind-atari-deep-q-learner An implementation of a deep reinforcement learning architecture for playing Atari games 1,826