1 repo
Awesome GitHub RepositoriesParallel Matrix Operations
Implementations of parallelized matrix multiplication for distributed hardware.
Distinguishing note: Focuses on custom parallel matrix multiplication.
Explore 1 awesome GitHub repository matching data & databases · Parallel Matrix Operations. Refine with filters or upvote what's useful.
Awesome Parallel Matrix Operations GitHub Repositories
Find the best repos with AI.We'll search the best matching repositories with AI.
- jax-ml/jax
jax-ml/jax
34,919View on GitHubThis project is a high-performance numerical computing library designed for large-scale scientific and machine learning workloads. It functions as an automatic differentiation framework and a just-in-time compilation engine, transforming high-level Python code into optimized machine instructions. By enforcing pure functional programming patterns and immutable array semantics, the library ensures that mathematical functions remain compatible with automated graph transformations and symbolic differentiation. The platform distinguishes itself through its distributed array computing capabilities,
Implements custom parallel matrix multiplication by sharding input tensors across devices.
Pythonjax
