1 repositorio
Use of mutable references to track neural network variables during initialization and debugging.
Distinct from Reference-Based Mutability Control: Specializes reference-based mutability control specifically for tracking model variables in a deep learning context.
Explore 1 awesome GitHub repository matching data & databases · Model Variable References. Refine with filters or upvote what's useful.
Flax is a deep learning framework and JAX neural network library designed for building complex machine learning models. It functions as a distributed training library and model state manager, providing a toolkit for defining flexible neural network architectures and scaling their training across multiple hardware devices. The project is characterized by a design that separates network logic from parameter values to remain compatible with pure functions. It uses hierarchical module composition to organize networks as trees of nested modules and employs a reference-based state management system
Uses mutable references to track model variables and state during initialization and debugging.