JAX
NumPy con autodiff, JIT y TPUs
Frameworks de ML & MLOpsComputación numéricaApache-2.0Ejecuta localmentePythonAvanzado
🐳 descargas de Docker
📦 descargas de PyPI / mes
📦 descargas de npm / mes
🚀 Última versión ·
Zumbido de la comunidad
📰 Publicaciones de Hacker News (30d) ·
🦋 Publicaciones de Bluesky esta semana
¿Qué es JAX?
JAX compone diferenciación automática, compilación JIT y vectorización — el sustrato para gran parte de la investigación de Google y DeepMind.
Por qué la gente elige JAX
- Compila a código rápido en GPU y TPU
- Diseño funcional que se compone de manera limpia
- Detrás de Gemma, MaxText y gran parte del trabajo de DeepMind
JAX en un vistazo
| Categoría | Frameworks de ML & MLOps |
| Tipo | Computación numérica |
| Licencia | Apache-2.0 |
| Ejecuta localmente | Sí |
| Construido con | Python |
| Nivel de habilidad | Avanzado |
| Mejor para | investigadores que quieren velocidad sin renunciar a la semántica de NumPy |
Últimas versiones
Alternativas de código abierto a JAX
Otros frameworks de ml de código abierto y herramientas de mlops que valen la pena comparar:
DagsterOrquestación que piensa en activos de datos, no en tareas
TensorFlowEl framework de aprendizaje profundo de Google, construido para producción
PyTorchEl framework en el que casi todos los modelos de IA modernos están escritos
OpenCVLa biblioteca de visión por computadora en la que se basa todo lo demás
scikit-learnAprendizaje automático clásico, hecho correctamente
Apache AirflowPrograma y monitorea pipelines de datos
RayEscala Python desde una laptop hasta un clúster
XGBoostSigue siendo el mejor en datos tabulares
Label StudioEtiqueta cualquier cosa: texto, imágenes, audio, video
MLflowRastrea experimentos y envía modelos sin la hoja de cálculo
ONNXMueve un modelo entre frameworks y entornos de ejecución
LightGBMAumento de gradiente que entrena rápido en tablas grandes
CVATAnotación seria para visión por computadora
DVCGit para conjuntos de datos y modelos
OptunaEncuentra los hiperparámetros correctos sin adivinarJAX 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
Preguntas frecuentes
¿Es JAX gratuito?
JAX es gratuito y de código abierto (licencia Apache-2.0), por lo que puedes usarlo, autoalojarlo y modificarlo sin costo alguno.
¿Puedo ejecutar JAX localmente?
Sí. JAX está diseñado para ejecutarse en tu propia máquina o servidor, manteniendo tus datos privados.
¿Cuál es la mejor alternativa a JAX?
Las alternativas populares de código abierto incluyen Dagster, TensorFlow, PyTorch. Consulta las comparaciones anteriores para elegir.
Explora más IA de código abierto
Navega por el directorio completo de herramientas, modelos y proyectos de IA de código abierto, actualizado diariamente.
Navega todas las herramientas →