Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Jun 11, 2024 - Python
Tensors and Dynamic neural networks in Python with strong GPU acceleration
AMD ROCm Performance Primitives (RPP) library is a comprehensive high-performance computer vision library for AMD processors with HIP/OpenCL/CPU back-ends.
Monte Carlo eXtreme for OpenCL (MCXCL)
Examples of programs built using Modal
Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
On-device AI across mobile, embedded and edge for PyTorch
WarpX is an advanced, time-based electromagnetic & electrostatic Particle-In-Cell code.
Pytorch domain library for recommendation systems
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Interactive data visualizations and plotting in Julia
SkyPilot: Run LLMs, AI, and Batch jobs on any cloud. Get maximum savings, highest GPU availability, and managed execution—all with a simple interface.
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
A model-independent chemistry module for atmosphere models
Safe rust wrapper around CUDA toolkit
CEED Library: Code for Efficient Extensible Discretizations
Add a description, image, and links to the gpu topic page so that developers can more easily learn about it.
To associate your repository with the gpu topic, visit your repo's landing page and select "manage topics."