JAX Basics - Framework Benchmarks & JIT Compilation

A deep dive into JAX fundamentals, comparing it with PyTorch and TensorFlow, and understanding JIT compilation.