oneflow

OneFlow é um framework de aprendizado profundo projetado para ser amigável, escalável e eficiente. Ele fornece uma API estilo PyTorch para programação de modelos, Tensor Global para escalar para execução paralela n-dimensional e um Compilador de Grafo para acelerar e implantar modelos. ### Principais Recursos - **API estilo PyTorch**: Permite programação intuitiva de modelos, semelhante ao PyTorch. - **Tensor Global**: Facilita a escalabilidade de modelos em arquiteturas paralelas multidimensionais. - **Compilador de Grafo**: Otimiza grafos computacionais para acelerar o treinamento e simplificar a implantação. ### Requisitos do Sistema - Sistema operacional Linux. - Python 3.7, 3.8, 3.9, 3.10 ou 3.11. - Arquitetura CUDA 6.0 ou superior, CUDA Toolkit 10.0 ou superior e driver Nvidia versão 440.33 ou superior para aceleração de GPU. ### Métodos de Instalação - **Docker**: Imagens Docker noturnas pré-construídas com suporte a CUDA estão disponíveis. - **Pip**: Versões estáveis e builds noturnos (somente CPU ou CUDA) podem ser instalados via pip, com opções de espelho doméstico para usuários na China. - **Código Fonte**: Usuários podem clonar o repositório e compilar usando CMake, com presets de configuração para ambientes somente CPU ou CUDA. ### Ecossistema e Model Zoo - **Libai**: Uma caixa de ferramentas projetada para treinamento paralelo de modelos transformer em larga escala, incluindo BERT, GPT, T5, VisionTransformer e SwinTransformer. - **FlowVision**: Uma caixa de ferramentas que fornece conjuntos de dados de visão computacional, modelos de última geração e utilitários. - **OneFlow-Models**: Fornece implementações de modelos como ResNet-50 e Wide&Deep (observado como desatualizado). ### Comunidade e Comunicação - **Canais em Chinês**: Grupo QQ (331883), WeChat (OneFlowXZS) e página da organização no Zhihu. - **Canais Internacionais**: Servidor Discord, Twitter, LinkedIn e blog Medium. - Relatórios de bugs e solicitações de recursos são gerenciados via issues do GitHub. OneFlow é originalmente desenvolvido pela OneFlow Inc. e Zhejiang Lab, e é distribuído sob a Licença Apache 2.0.