mindspore  by mindspore-ai

Deep learning framework for mobile, edge, and cloud training/inference

created 5 years ago
4,575 stars

Top 10.9% on sourcepulse

GitHubView on GitHub
Project Summary

MindSpore is an open-source deep learning framework designed for mobile, edge, and cloud scenarios, targeting data scientists and algorithmic engineers. It offers native support for Ascend AI processors and aims for software-hardware co-optimization, providing a friendly development experience and efficient execution.

How It Works

MindSpore employs Source Transformation (ST) for automatic differentiation, contrasting with Operator Overloading (OO) used by frameworks like PyTorch. ST enables static compilation optimization and handles complex control flows natively, leading to potential performance gains. Its automatic parallelization supports data, model, and hybrid parallelism through fine-grained operator splitting, abstracting complexity from the user.

Quick Start & Requirements

  • Installation: Primarily via pip with pre-built wheels or by compiling from source. Docker images are also available.
  • Dependencies: Supports Ascend, NVIDIA GPU (CUDA 10.1), and CPU. Specific CUDA versions may be required for GPU builds.
  • Resources: Installation details and build options for various platforms are available in the installation guide.

Highlighted Details

  • Native support for Ascend AI processors.
  • Automatic differentiation via Source Transformation (ST) for performance and control flow handling.
  • Automatic parallelization for distributed training.
  • Available via pip, source compilation, and Docker images.

Maintenance & Community

  • Active maintenance with multiple branches (e.g., r2.2, r2.1, r2.0 are "Maintained"). Older branches are marked "End Of Life".
  • Community communication via Slack. Contribution guidelines are available.

Licensing & Compatibility

  • Licensed under the Apache License 2.0.
  • Permissive license suitable for commercial use and integration with closed-source projects.

Limitations & Caveats

  • GPU support is specified for CUDA 10.1, which may be outdated. Newer CUDA versions might require source compilation.
  • The maintenance status indicates a clear lifecycle for branches, with older versions reaching End Of Life.
Health Check
Last commit

1 year ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
3
Star History
110 stars in the last 90 days

Explore Similar Projects

Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Georgios Konstantopoulos Georgios Konstantopoulos(CTO, General Partner at Paradigm), and
7 more.

ThunderKittens by HazyResearch

0.6%
3k
CUDA kernel framework for fast deep learning primitives
created 1 year ago
updated 3 days ago
Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Nat Friedman Nat Friedman(Former CEO of GitHub), and
32 more.

llama.cpp by ggml-org

0.4%
84k
C/C++ library for local LLM inference
created 2 years ago
updated 16 hours ago
Feedback? Help us improve.