آموزش یادگیری JAX

سرفصل ها و درس ها

مقدمه Introduction

• پیش نیازها Prerequisites

• JAX چیست؟ What is JAX?

• چرا از JAX استفاده کنیم؟ Why use JAX?

• انتخاب JAX Choosing JAX

• JAX در مقابل TensorFlow در مقابل PyTorch JAX vs. TensorFlow vs. PyTorch

• راه اندازی با Colab Getting set up with Colab

• امتحان فصل Chapter Quiz

1. کار با آرایه های JAX 1. Working with JAX Arrays

• آرایه های JAX JAX arrays

• آرایه های JAX و آرایه های NumPy: شباهت ها JAX arrays and NumPy arrays: Similarities

• آرایه های JAX و آرایه های NumPy: تفاوت ها JAX arrays and NumPy arrays: Differences

• سرعت ارسال ناهمزمان و آرایه JAX افزایش می یابد Asynchronous dispatch and JAX array speed up

• امتحان فصل Chapter Quiz

2. تالیف Just-in-Time 2. Just-in-Time Compilation

• تبدیل تابع قابل ترکیب Composable function transformations

• JIT و توابع خالص JIT and pure functions

• با استفاده از JIT Using JIT

• اشیاء ردیاب در JIT Tracer objects in JIT

• توابع ناخالص و JIT: جریان های ورودی/خروجی Impure functions and JIT: I/O streams

• توابع ناخالص و JIT: حالت جهانی Impure functions and JIT: Global state

• توابع ناخالص و JIT: تکرار کننده ها Impure functions and JIT: Iterators

• Jaxprs Jaxprs

• کنترل بیانیه های جریان و JIT Control flow statements and JIT

• آرگومان های ایستا در توابع جیت شده Static arguments in jitted functions

• Lambdas و JIT Lambdas and JIT

• امتحان فصل Chapter Quiz

3. بردارسازی خودکار و موازی سازی خودکار 3. Automatic Vectorization and Automatic Parallelization

• درک برداری و موازی سازی Understanding vectorization and parallelization

• بردارسازی خودکار Automatic vectorization

• مقایسه دسته بندی ساده و دستی با برداری خودکار Comparing naive and manual batching with automatic vectorization

• امتحان فصل Chapter Quiz

4. محاسبه گرادیان 4. Computing Gradients

• درک محاسبات گرادیان Understanding gradient computation

• محاسبه گرادیان Gradient computation

• شیب مرتبه بالاتر Higher order gradients

• یعقوبیان و هسی ها Jacobians and Hessians

5. انجام تبدیل بر روی pytrees 5. Performing Transformations on pytrees

• درک pytrees Understanding pytrees

• pytrees ساده Simple pytrees

• عملیات روی pytrees Operations on pytrees

• ظروف Pytree Pytree containers

• ظروف سفارشی به عنوان pytrees Custom containers as pytrees

• امتحان فصل Chapter Quiz

6. آموزش شبکه های عصبی ساده 6. Training Simple Neural Networks

• رگرسیون با استفاده از یک نورون واحد: بارگذاری و پیش پردازش داده ها Regression using a single neuron: Loading and preprocessing data

• رگرسیون با استفاده از یک نورون واحد: توابع کمک کننده Regression using a single neuron: Helper functions

• رگرسیون با استفاده از یک نورون واحد: آموزش و ارزیابی یک مدل Regression using a single neuron: Training and evaluating a model

• رگرسیون با استفاده از شبکه عصبی: توابع کمکی Regression using a neural network: Helper functions

• رگرسیون با استفاده از شبکه عصبی: آموزش یک مدل و تجسم نتایج Regression using a neural network: Training a model and visualizing results

• طبقه بندی با استفاده از شبکه عصبی: بارگذاری و پیش پردازش داده ها Classification using neural network: Loading and preprocessing data

• طبقه بندی با استفاده از شبکه عصبی: مدل آموزش و ارزیابی Classification using neural network: Training and evaluating model

نتیجه Conclusion

• خلاصه و مراحل بعدی Summary and next steps