• 

  JAX: Numpy with Gradients, GPUs and TPUs

  

  github.com

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  python repository

  • Add  

  2

More like this (3)

• Nx (Numerical Elixir) is now publicly available

  

  dashbit.co

  • Add  

  
• 

  What I Wish Someone Had Told Me About Tensor Computation Libraries

  

  eigenfoo.xyz

  • Read
  • Add  

  
• 

  Flax: Google’s Open Source Approach To Flexibility In Machine Learning

  

  hackernoon.com

  • Read
  • Add