mikelaud / ai-jax

ai-jax

ai-jax

1. Intro to JAX: Accelerating Machine Learning research (Jake VanderPlas)
2. Install JAX:
   • Ubuntu-20.04
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.7
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.8
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.9
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.10
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.11
     • Install WSL2 + Ubuntu-20.04 + CUDA-12 + JAX-0.4.13
   • Ubuntu-22.04
     • Install WSL2 + Ubuntu-22.04 + CUDA-12 + JAX-0.4.7
     • Install WSL2 + Ubuntu-22.04 + CUDA-12 + JAX-0.4.8
     • Install WSL2 + Ubuntu-22.04 + CUDA-12 + JAX-0.4.9
     • Install WSL2 + Ubuntu-22.04 + CUDA-12 + JAX-0.4.13
3. JAX documentation
4. LinkedIn Learning: Learning JAX (Janani Ravi)
5. Machine Learning with JAX (Aleksa Gordić)
   • (Tutorial 1) ML with JAX: From Zero to Hero
   • (Tutorial 2) ML with JAX: From Hero to HeroPro+
   • (Tutorial 3) ML with JAX: Coding a Neural Network from Scratch in Pure JAX
6. JAX Crash Course (AssemblyAI)