JAX
NumPy com autodiff, JIT e TPUs
Frameworks de ML & MLOpsComputação numéricaApache-2.0Executa localmentePythonAvançado
🐳 pulls do Docker
📦 downloads do PyPI / mês
📦 downloads do npm / mês
🚀 Última versão ·
O que é JAX?
JAX compõe diferenciação automática, compilação JIT e vetorização — a base para grande parte da pesquisa do Google e da DeepMind.
Por que as pessoas escolhem JAX
- Compila para código rápido em GPU e TPU
- Design funcional que se compõe de forma limpa
- Por trás do Gemma, MaxText e muito do trabalho da DeepMind
JAX em um relance
| Categoria | Frameworks de ML & MLOps |
| Tipo | Computação numérica |
| Licença | Apache-2.0 |
| Executa localmente | Sim |
| Construído com | Python |
| Nível de habilidade | Avançado |
| Melhor para | pesquisadores que querem velocidade sem abrir mão da semântica do NumPy |
Últimas versões
Alternativas de código aberto ao JAX
Outros frameworks de ml de código aberto & ferramentas de mlops que valem a pena comparar:
DagsterOrquestração que pensa em ativos de dados, não em tarefas
TensorFlowO framework de deep learning do Google, construído para produção
PyTorchO framework em que quase todos os modelos de IA modernos são escritos
OpenCVA biblioteca de visão computacional sobre a qual tudo mais é construído
scikit-learnAprendizado de máquina clássico, feito corretamente
Apache AirflowAgende e monitore pipelines de dados
RayEscale Python de um laptop para um cluster
XGBoostAinda é o melhor em dados tabulares
Label StudioRotule qualquer coisa — texto, imagens, áudio, vídeo
MLflowAcompanhe experimentos e envie modelos sem a planilha
ONNXMova um modelo entre frameworks e tempos de execução
LightGBMAumento de gradiente que treina rapidamente em grandes tabelas
CVATAnotação séria para visão computacional
DVCGit para conjuntos de dados e modelos
OptunaEncontre os hiperparâmetros certos sem adivinhaçãoJAX cara a cara
JAX vs DagsterJAX vs TensorFlowJAX vs PyTorchJAX vs OpenCVJAX vs scikit-learnJAX vs Apache AirflowJAX vs RayJAX vs XGBoostJAX vs Label StudioJAX vs MLflowJAX vs ONNXJAX vs LightGBMJAX vs CVATJAX vs DVCJAX vs Optuna
FAQ
O JAX é gratuito?
O JAX é gratuito e de código aberto (licença Apache-2.0), então você pode usá-lo, hospedá-lo e modificá-lo sem custo.
Posso executar o JAX localmente?
Sim. O JAX é projetado para rodar na sua própria máquina ou servidor, mantendo seus dados privados.
Qual é a melhor alternativa ao JAX?
Alternativas populares de código aberto incluem Dagster, TensorFlow, PyTorch. Veja as comparações acima para escolher.
Explore mais IA de código aberto
Navegue pelo diretório completo de ferramentas, modelos e projetos de IA de código aberto — atualizado diariamente.
Navegue por todas as ferramentas →