1 repo
Awesome GitHub RepositoriesDistributed Parallelism APIs
APIs for mapping functions over data shards across multiple hardware devices.
Distinguishing note: Focuses on the API for distributed device management, distinct from the underlying topology configuration.
Explore 1 awesome GitHub repository matching devops & infrastructure · Distributed Parallelism APIs. Refine with filters or upvote what's useful.
Awesome Distributed Parallelism APIs 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,
Distributes computations across multiple hardware units by mapping functions over data shards and managing device meshes.
Pythonjax
