jax

A list of content tagged jax

Responses

• Introducing Metrax: performant, efficient, and robust model evaluation metrics in JAX
• Google Penzai

Bookmarks

• Introducing Keras Core: Keras for TensorFlow, JAX, and PyTorch.