pytorch-jax-implementation introduction to pytorch(activation_function,introduction) jax tutorial for deep learning(introduction, activation_function)