TensorFlow para Java

Veja no TensorFlow.org Veja o repositório do GitHub

O TensorFlow Java pode ser executado em qualquer JVM para criar, treinar e executar modelos de aprendizado de máquina. Ele vem com uma série de utilitários e estruturas que ajudam a realizar a maioria das tarefas comuns aos cientistas de dados e desenvolvedores que trabalham nesse domínio. Java e outras linguagens JVM, como Scala ou Kotlin, são frequentemente usadas em empresas de pequeno a grande porte em todo o mundo, o que torna o TensorFlow uma escolha estratégica para adotar o aprendizado de máquina em grande escala.

O Repositório

No início, as vinculações da linguagem Java para o TensorFlow eram hospedadas no repositório principal do TensorFlow e lançadas apenas quando uma nova versão da biblioteca principal estava pronta para ser distribuída, o que acontece apenas algumas vezes por ano. Agora, todo o código relacionado a Java foi movido para este repositório para que possa evoluir e ser lançado independentemente das versões oficiais do TensorFlow. Além disso, a maioria das tarefas de construção foi migrada do Bazel para o Maven, que é mais familiar para a maioria dos desenvolvedores Java.

O seguinte descreve o layout do repositório e seus diferentes artefatos:

  • tensorflow-core

    • Todos os artefatos que constroem as principais ligações de linguagem do TensorFlow para Java
    • Público-alvo: projetos que fornecem suas próprias APIs ou estruturas no TensorFlow e querem apenas uma camada fina para acessar o tempo de execução do TensorFlow a partir da JVM
  • tensorflow-framework

    • API primária para construir e treinar redes neurais com o TensorFlow
    • Público-alvo: desenvolvedores de redes neurais
  • ndarray

    • Biblioteca de utilitários genéricos para operações de E/S de dados n-dimensionais
    • Usado pelo TensorFlow, mas não depende do TensorFlow
    • Público-alvo: qualquer desenvolvedor que precise de uma implementação de array n-dimensional Java, usando ou não com o TensorFlow

Comunicação

Este repositório é mantido pelo TensorFlow JVM Special Interest Group (SIG). Você pode facilmente entrar no grupo assinando a lista de discussão jvm@tensorflow.org , ou pode simplesmente enviar pull requests e levantar problemas para este repositório. Há também um canal Gitter sig-jvm .