ÁLVAROMAUREIRA

Keras 3: Rediseño Completo del Framework de Aprendizaje Profundo Multi-Backend para JAX, PyTorch y TensorFlow

Publicado ·

El equipo de desarrollo de Keras ha anunciado el lanzamiento oficial de Keras 3, una reescritura arquitectónica completa del icónico framework de aprendizaje profundo que introduce compatibilidad multi-backend agnóstica para JAX, PyTorch y TensorFlow sin requerir modificaciones en el código de modelado.

Convergencia de Ecosistemas en un Framework Único

Durante años, los ingenieros e investigadores de inteligencia artificial se vieron forzados a elegir entre ecosistemas cerrados: la madurez productiva y despliegue móvil de TensorFlow, la flexibilidad y dinamismo de PyTorch, o las transformaciones computacionales de alta velocidad y compilación XLA de JAX. Keras 3 elimina esta fragmentación al desacoplar completamente la API de alto nivel del motor de ejecución de tensores subyacente.

Con Keras 3, un mismo bloque o modelo neuronal definido con capas estándar de Keras puede ser instanciado y entrenado sobre tensores de PyTorch utilizando optimizadores nativos de PyTorch, o compilado mediante jax.jit para ejecución vectorizada en clústeres de TPUs y GPUs de última generación.

Los desarrolladores pueden consultar las guías de migración y la documentación completa en el portal oficial de Keras 3.

Compilación XLA, Distribución de Modelos y Rendimiento

Además de la interoperabilidad entre frameworks, Keras 3 introduce un espacio de nombres integral de operaciones matemáticas de bajo nivel (keras.ops), que permite escribir capas y funciones de pérdida personalizadas totalmente portables que se compilan nativamente al backend seleccionado.

La nueva arquitectura ofrece un rendimiento superior gracias a la optimización de grafos estáticos vía XLA, acelerando el entrenamiento distribuido con paralelismo de datos y de tensores en grandes infraestructuras de supercomputación.

Sigue aprendiendo

Explora los recursos y continúa tu recorrido en la comunidad.

Únete gratis a la comunidad →