91 packages found

mamba-ssm

Mamba state-space model
  1. cuda
  2. pytorch
  3. state-space
  4. model
34 Contributors
2.2.4published 4 months agoOther

vllm

A high-throughput and memory-efficient inference and serving engine for LLMs
  1. amd
  2. cuda
  3. deepseek
  4. gpt
  5. hpu
  6. inference
  7. inferentia
  8. llama
  9. llm
  10. llm-serving
  11. llmops
  12. mlops
  13. model-serving
  14. pytorch
  15. qwen
  16. rocm
  17. tpu
  18. trainium
  19. transformer
  20. xpu
397 Contributors
0.8.3published 5 days agoOther

nipet

CUDA-accelerated Python utilities for high-throughput PET/MR image reconstruction and analysis
  1. PET
  2. image
  3. reconstruction
  4. analysis
  5. cuda
  6. gpu
  7. image-reconstruction
  8. medical-imaging
  9. mlem
  10. processing
  11. python
2.2.0published 2 years agoApache-2.0

rmm-cu11

rmm - RAPIDS Memory Manager
  1. cuda
  2. memory-allocation
  3. memory-management
  4. rapids
91 Contributors
25.2.0published 2 months agoApache-2.0

cucim-cu12

cuCIM - an extensible toolkit designed to provide GPU accelerated I/O, computer vision & image processing primitives for N-Dimensional images with a focus on biomedical imaging.
  1. computer-vision
  2. cuda
  3. digital-pathology
  4. gpu
  5. image-analysis
  6. image-data
  7. image-processing
  8. medical-imaging
  9. microscopy
  10. multidimensional-image-processing
  11. nvidia
  12. segmentation
31 Contributors
25.2.0published 2 months agoApache-2.0

cupy-cuda115

CuPy: NumPy & SciPy for GPU
  1. cublas
  2. cuda
  3. cudnn
  4. cupy
  5. curand
  6. cusolver
  7. cusparse
  8. cusparselt
  9. cutensor
  10. gpu
  11. nccl
  12. numpy
  13. nvrtc
  14. nvtx
  15. python
  16. rocm
  17. scipy
  18. tensor
339 Contributors
10.6.0published 3 years agoMIT

dace

Data-Centric Parallel Programming Framework
  1. cuda
  2. fpga
  3. high-level-synthesis
  4. high-performance-computing
  5. programming-language
  6. vivado-hls
67 Contributors
1.0.2published 3 weeks agoBSD-3-Clause

torchao

Package for applying ao techniques to GPU models
  1. brrr
  2. cuda
  3. dtypes
  4. float8
  5. inference
  6. llama
  7. mx
  8. offloading
  9. optimizer
  10. pytorch
  11. quantization
  12. sparsity
  13. training
  14. transformer
0.10.0published 3 days agoBSD-3-Clause

cupy-cuda110

CuPy: NumPy & SciPy for GPU
  1. cublas
  2. cuda
  3. cudnn
  4. cupy
  5. curand
  6. cusolver
  7. cusparse
  8. cusparselt
  9. cutensor
  10. gpu
  11. nccl
  12. numpy
  13. nvrtc
  14. nvtx
  15. python
  16. rocm
  17. scipy
  18. tensor
321 Contributors
12.3.0published 1 year agoMIT

cupy-rocm-5-0

CuPy: NumPy & SciPy for GPU
  1. cublas
  2. cuda
  3. cudnn
  4. cupy
  5. curand
  6. cusolver
  7. cusparse
  8. cusparselt
  9. cutensor
  10. gpu
  11. nccl
  12. numpy
  13. nvrtc
  14. nvtx
  15. python
  16. rocm
  17. scipy
  18. tensor
321 Contributors
13.3.0published 8 months agoMIT
Showing 1 to 10 of 91 results