Descrição
Redes neurais podem ser criadas em JAX com abstrações de módulos do DeepMind Haiku. Esta biblioteca é para pesquisadores e desenvolvedores de aprendizado de máquina que desenham arquiteturas de modelos. Treino pode consumir muita computação e datasets podem conter registros sensíveis.