#pytorch

  1. safetensors

    functions to read and write safetensors which aim to be safer than their PyTorch counterpart. The format is 8 bytes which is an unsized int, being the size of a JSON header, the JSON…

    v0.7.0 640K #tensor #pytorch #hugging-face #tensorflow
  2. burn

    Flexible and Comprehensive Deep Learning Framework in Rust

    v0.20.1 53K #deep-learning #machine-learning #pytorch #multi-dimensional-array
  3. tch

    Rust wrappers for the PyTorch C++ api (libtorch)

    v0.23.0 96K #deep-learning #pytorch #machine-learning
  4. torch-sys

    Low-level FFI bindings for the PyTorch C++ api (libtorch)

    v0.23.0 95K #deep-learning #pytorch #machine-learning #ffi
  5. mel_spec

    Mel spectrograms aligned to the results from the whisper.cpp, pytorch and librosa reference implementations and suited to streaming audio

    v0.3.4 34K #audio #audio-streaming #mel #whisper-cpp #pytorch #spectrograms #tga #spectrogram #librosa #save-image
  6. rten

    Machine learning runtime

    v0.24.0 1.6K #onnx #machine-learning-models #inference-engine #tensor #operator #model-format #pytorch #neural-network
  7. torsh-linalg

    Linear algebra operations for ToRSh with SciRS2 integration

    v0.1.0-rc.1 #solver #tensor #torsh #matrix-inverse #algebra #sci-rs2 #least-squares #determinant #cholesky #pytorch
  8. mosec

    Model Serving made Efficient in the Cloud

    v0.9.0 2.7K #deployment #model-serving #deep-learning #machine-learning-models #inference #gpu #pytorch #model-deployment #graceful-shutdown #llm
  9. build2cmake

    Generate CMake files for kernel-builder projects

    v0.12.1 #kernel #kernel-builder #pytorch #machine-learning #generate #python-packages
  10. rustorch

    Production-ready PyTorch-compatible deep learning library in Rust with special mathematical functions (gamma, Bessel, error functions), statistical distributions, Fourier transforms (FFT/RFFT)…

    v0.6.29 6.9K #pytorch #matrix-decomposition #fft #machine-learning
  11. torsh-data

    Data loading and preprocessing utilities for ToRSh

    v0.1.0-rc.1 #sampler #torsh #data-loader #pre-processor #sampling-strategies #data-loading #dataframe #audio #gpu-acceleration #pytorch
  12. torsh-nn

    Neural network modules for ToRSh with PyTorch-compatible API

    v0.1.0-rc.1 #neural-network #torsh #deep-learning #machine-learning #initialization #pytorch #gelu #building-block #sci-rs2
  13. torsh-autograd

    Automatic differentiation engine for ToRSh with PyTorch-compatible API

    v0.1.0-rc.1 #anomaly-detection #automatic-differentiation #performance-profiling #operation #memory-optimization #torsh #checkpointing #pytorch #sci-rs2 #jacobi
  14. torsh-tensor

    Tensor implementation for ToRSh with PyTorch-compatible API

    v0.1.0-rc.1 230 #deep-learning #operation #neural-network #torsh #automatic-differentiation #sci-rs2 #pytorch #broadcasting #machine-learning
  15. executorch

    Rust bindings for ExecuTorch - On-device AI across mobile, embedded and edge for PyTorch

    v0.8.1 #pytorch #bindings #machine-learning #edge-device #no-alloc
  16. burn-import

    importing datamodels into the Burn framework

    v0.20.1 4.7K #onnx #safetensors #burn-framework #pytorch #machine-learning-models #serialization #neural-network
  17. axonml

    A complete ML/AI framework in pure Rust - PyTorch-equivalent functionality

    v0.2.8 #artificial-intelligence #inference #dataset #pytorch #training #ml #audio #classification #distributed #learning-rate
  18. tlparse

    Parse TORCH_LOG logs produced by PyTorch torch.compile

    v0.4.6 850 #logging #pytorch #log-parser #compile #process
  19. datasynth-graph

    Graph/network export for synthetic accounting data - supports PyTorch Geometric, Neo4j, and DGL formats

    v0.4.1 #accounting #graph #pytorch #exporter #synthetic-data #neo4j #hypergraph #dgl #ml #multi-layer
  20. gpt_sovits_rs

    A GPT_Sovits inference lib in Rust base on pytorch

    v0.7.4 2.3K #pytorch #speech-synthesis #model #inference #convert
  21. mini-ode

    A minimalistic ODE solvers library built on top of PyTorch

    v0.1.4 #pytorch #ode-solver #optimization #built #top
  22. tensor-man

    A small utility to inspect and validate safetensors and ONNX files

    v0.4.2 290 #safetensors #gguf #onnx #machine-learning-models #sign-verify #docker #pytorch
  23. cryptotensors

    extension of safetensors that adds encryption, signing, and access control (Rego-based policy engine) while maintaining full backward compatibility with the safetensors format. It…

    v0.2.0 #encryption #safetensors #pytorch #security
  24. burn-tensor

    Tensor library with user-friendly APIs and automatic differentiation support

    v0.20.1 53K #deep-learning #pytorch #machine-learning #multi-dimensional-array
  25. torch_poetry_bootstrap

    A command-line tool to detect CUDA version and install the appropriate PyTorch wheel via Poetry

    v0.1.18 190 #pytorch #poetry #cuda #bootstrap
  26. bintensors

    high-performance binary tensor serialization format designed to be faster eliminating use of JSON serialization metadata

    v0.1.1 370 #tensor #hugging-face #pytorch #serde
  27. kernel-abi-check

    Check the ABI of Hub Kernels

    v0.12.1 #kernel #abi #machine-learning #hub #build #python-packages #kernel-builder #pytorch #nix
  28. burn-train

    Training crate for the Burn framework

    v0.20.1 32K #deep-learning #pytorch #machine-learning #multi-dimensional-array
  29. pyo3-dlpack

    Zero-copy DLPack tensor interop for PyO3

    v0.1.0 #pytorch #dlpack #pyo3 #numpy #tensor
  30. deep_quokka

    Unofficial deep learning library

    v0.1.17 900 #deep-learning #model #multidimensional-array #onnx #있습니다 #wgpu #multi-dimensional-array #web-gpu #candle #pytorch
  31. diffai

    AI/ML model diff CLI - PyTorch, Safetensors, NumPy, MATLAB comparison

    v0.4.1 #safetensors #diff #ml #pytorch
  32. nutorch

    wrapper for tch-rs, itself a wrapper for libtorch, the C++ backend for PyTorch. It provides a plugin for Nushell to use PyTorch tensors on the command line.

    v0.1.4 #pytorch #tensor #nu-shell-plugin #command-line #back-end #libtorch #tch-rs #garbage-collection
  33. medrs

    Ultra-high-performance medical imaging I/O for deep learning

    v0.1.2 #deep-learning #pytorch #medical-imaging #nifti #neuroimaging
  34. torsh-functional

    Functional programming utilities for ToRSh tensors

    v0.1.0-rc.1 #deep-learning #matrix #torsh #element-wise #neural-network #artificial-intelligence #math-operations #signal-processing #pytorch #convolution
  35. torsh-cluster

    Unsupervised learning and clustering algorithms for ToRSh, powered by SciRS2

    v0.1.0-rc.1 #cluster-analysis #torsh #deep-learning #unsupervised-learning #dbscan #k-means #sci-rs2 #spectral #pytorch #evaluation-metrics
  36. burn-nn

    Neural network building blocks for the Burn deep learning framework

    v0.20.1 16K #deep-learning #pytorch #machine-learning
  37. torsh-benches

    Benchmarking suite for ToRSh

    v0.1.0-rc.1 #benchmark #performance-analysis #deep-learning #matrix #torsh #performance-profiling #matrix-multiplication #analysis-optimization #neural-network #pytorch
  38. torsh-python

    Python bindings for ToRSh - PyTorch-compatible deep learning in Rust

    v0.1.0-rc.1 #deep-learning #python-bindings #torsh #pytorch #devices #neural-network
  39. torsh-vision

    Computer vision utilities for ToRSh deep learning framework

    v0.1.0-rc.1 #deep-learning #computer-vision #pytorch #machine-learning
  40. burn-backend

    Core backend interfaces and data structures for executing tensor operations in Burn

    v0.20.1 3.3K #tensor #deep-learning #pytorch #machine-learning
  41. torsh-text

    Natural language processing utilities for ToRSh deep learning framework

    v0.1.0-rc.1 #deep-learning #pytorch #machine-learning
  42. burn-optim

    Optimizer building blocks for the Burn deep learning framework

    v0.20.1 16K #deep-learning #pytorch #machine-learning #multi-dimensional-array
  43. torsh-signal

    Signal processing utilities for ToRSh with SciRS2 integration

    v0.1.0-rc.1 #signal-processing #torsh #operation #deep-learning #sci-rs2 #pytorch #stft #spectral #processing-algorithm #convolution
  44. candle-crf

    porting pytorch-crf to candle

    v0.8.8 550 #crf #candle #pytorch #pytorch-crf
  45. deep_filter

    Noise supression using deep filtering

    v0.2.5 #speech #audio #speech-enhancement #dataset #pytorch #audio-processing #embedded-devices #real-time-audio-processing #audio-samples
  46. torsh-package

    Model packaging and distribution utilities for ToRSh

    v0.1.0-rc.1 #packaging #package-metadata #model #torsh #package-versioning #pytorch #deep-learning #packaging-distribution #package-format #version-compatibility
  47. tch-plus

    Rust wrappers for the PyTorch C++ api (libtorch)

    v0.18.3 #pytorch #deep-learning #machine-learning
  48. ai-dataloader

    PyTorch DataLoader

    v0.6.2 700 #data-loader #pytorch #tensorflow #machine-learning
  49. rust-lstm-1025

    BMI (Basic Model Interface) adapter for LSTM-based streamflow prediction

    v0.1.1 #lstm #interface-adapter #model #bmi #python #pytorch #streamflow #next-gen
  50. pyo3-tch

    Manipulate PyTorch tensors from a Python extension via PyO3/tch

    v0.23.0 950 #deep-learning #pytorch #machine-learning
  51. diffusers

    using Torch

    v0.3.1 #pytorch #deep-learning #machine-learning #diffusion #transformer
  52. silero_vad_jit

    A Silero_vad inference lib in Rust base on pytorch

    v0.2.0 #text-to-speech #pytorch #silero-vad #audio-processing #inference #silence #voice-activity-detection #python-version
  53. executorch-sys

    Unsafe Rust bindings for ExecuTorch - On-device AI across mobile, embedded and edge for PyTorch

    v0.8.1 #pytorch #edge-device #machine-learning #bindings #no-alloc
  54. convolutions-rs

    Fast, minimal dependency, completely Rust implementation of convolutions for machine learning

    v0.3.4 #machine-learning #convolution #pytorch #completely #transposed #neural-network #unit-testing #benchmark
  55. kaffe

    Pytorch, but make it rust :)

    v0.2.0 #pytorch #numpy #tensorflow #matrix
  56. nnsplit

    split text using a neural network. For sentence boundary detection, compound splitting and more.

    v0.5.9 #deep-learning #machine-learning #pytorch #sentencizer
  57. torch-sys-plus

    Low-level FFI bindings for the PyTorch C++ api (libtorch)

    v0.18.2 #deep-learning #pytorch #machine-learning #ffi
  58. Try searching with DuckDuckGo.

  59. rstorch

    from scratch of a neural network framework in Rust inspired by PyTorch

    v0.2.0 #deep-learning #pytorch #machine-learning #neural-network #rust
  60. chela

    High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust

    v0.0.2 #machine-learning #numpy #linear-algebra #pytorch #vector
  61. relearn

    A Reinforcement Learning library

    v0.3.1 #reinforcement-learning #action #agent #optimization #actor #pytorch #tch #tabular
  62. surya

    multilingual document OCR toolkit, original implementation in Python and PyTorch

    v0.4.0 #ocr #pytorch #multilingual #toolkit #original #opencv
  63. burn-core

    Flexible and Comprehensive Deep Learning Framework in Rust

    v0.20.1 54K #deep-learning #pytorch #machine-learning #multi-dimensional-array
  64. candle-ext

    An extension library to Candle that provides PyTorch functions not currently available in Candle

    v0.1.7 #candle #pytorch #extension #tensor #devices #dot-product #d-type #outer #attention #eye
  65. mpnet-rs

    translation of MPNet from PyTorch into Rust Candle

    v0.1.3 280 #safetensors #pytorch #candle
  66. torsh-core

    Core types and traits for ToRSh deep learning framework

    v0.1.0-rc.1 240 #deep-learning #devices #web-gpu #cuda #operation #stride #broadcasting #pytorch #sci-rs2 #neural-network
  67. candle-birnn

    implement Pytorch LSTM and BiDirectional LSTM with Candle

    v0.2.3 360 #lstm #pytorch #candle
  68. diffai-core

    Core library for AI/ML diff analysis - PyTorch, Safetensors, tensor statistics

    v0.5.0 #safetensors #ml #pytorch #diff
  69. flowrider

    High-performance PyTorch-compatible streaming dataset with distributed caching for on-the-fly remote dataset fetching

    v0.1.1 #stream #distributed #pytorch #cache #dataset #fetching #stream-data #on-the-fly #cache-data #cloud-storage
  70. tch-ext

    Sample Python extension using tch to interact with PyTorch

    v0.2.0 190 #pytorch #deep-learning #machine-learning
  71. candle-lstm

    optimize HuggingFace candle LSTM in some cases

    v0.2.3 320 #lstm #candle #pytorch
  72. anime-seg-rs

    Trying to implement anime-seg in Rust

    v0.3.0 #image-segmentation #anime #deep-learning #pytorch
  73. blowtorch

    running and creating Rust models trained in Pytorch

    v0.1.1 #pytorch #training #model #machine-learning #neural-network #compression
  74. tinygrad

    You like pytorch? You like micrograd? You love tinygrad! ❤️

    v0.1.0 #deep-learning #pytorch #machine-learning
  75. aotinductor

    Rust wrappers for the PyTorch AOTInductor api

    v0.1.2 #pytorch #deep-learning #machine-learning
  76. tboard

    Write and read tensorboard files

    v0.1.1 #tensorboard #deep-learning #pytorch
  77. ckpt-analyzer

    fetching the name and short hash for PyTorch ckpt or Safetensors model files

    v1.0.1 #safetensors #model #fetching #hash #pytorch
  78. rustorch-macros

    Procedural macros for RusTorch tensor creation

    v0.6.29 #tensor #pytorch #macro
  79. condor

    ergonomic wrapper around the tch library for Rust

    v0.1.6 #tch #screenshot #pytorch #interface
  80. nanograd

    A nano neural network library. Scalar valued types with a PyTorch like API.

    v0.1.2 #neural-network #scalar-valued #networking #api #pytorch #video #andrej #karpathy
  81. onnx-ir-derive

    Derive macros for onnx-ir

    v0.20.1 250 #onnx #deep-learning #web-gpu #burn #automatic-differentiation #macro-derive #safetensors #neural-network #pytorch #kernel-fusion
  82. tboard-pyo3

    Write and read tensorboard files

    v0.1.1 #tensorboard #pytorch #deep-learning
  83. siren-torch

    SIREN neural network implementation in Rust for PyTorch

    v0.1.1 #deep-learning #pytorch #signal-processing #machine-learning
  84. ndslice

    data structures to support n-d arrays of ranks

    v0.0.0 #selection #shape #array #pytorch #ranks #compact-representation #multi-dimensional #algebra #proc-macro #n-d