Thành phố Cuồng loạn

JAX là framework cho phép viết mã Python có khả năng biên dịch sang XLA, tối ưu cho cả CPU và TPU. Dưới đây là cách xây dựng một mạng nơ-ron đa tầng để phân loại ảnh chữ số viết tay. Cấu trúc mạng nơ-ron Mạng gồm ba phần chính: hàm khởi tạo trọng số, hàm truyền tiếp (forward pass), và cơ chế tối ưu gradient. JAX sử dụng cơ chế pure function, ...