GitHub

penzai (Models)

A JAX research toolkit for building, editing, and visualizing neural networks. | Python | ⭐ 2,000 | Models

1 articles in archive

[Models] google-deepmind/penzai — 2.0k⭐

A JAX research toolkit for building, editing, and visualizing neural networks. Language: Python | Stars: 2.0k | Category: Models Repository: https://github.com/google-deepmind/penzai

GitHub1d ago