NNX – Neural Networks for JAX | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

		NNX – Neural Networks for JAX (github.com/cgarciae)
		2 points by cgarciae on June 12, 2023 \| hide \| past \| favorite \| 1 comment

cgarciae on June 12, 2023 [–]

NNX is a Neural Networks library for JAX that provides a simple yet powerful module system that adheres to standard Python semantics. Its aim is to combine the robustness of Flax with a simplified, Pythonic API akin to that of PyTorch.

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact