Skip to content
#

Deep learning

Deep learning is an AI function and subset of machine learning, used for processing large amounts of complex data.

Here are 30,833 public repositories matching this topic...

kurtamohler
kurtamohler commented Jan 20, 2021

🚀 Feature

Add support for torch.max with:

  • CUDA bfloat16
  • CPU float16 and bfloat16

Motivation

Currently, torch.max has support for CUDA float16:

>>> torch.rand(10, dtype=torch.float16, device='cuda').max()
tensor(0.8530, device='cuda:0', dtype=torch.float16)

But all three other combinations of CPU/CUDA and float16/bfloat16 are not supported:

>>> torch.ra

Data science Python notebooks: Deep learning (TensorFlow, Theano, Caffe, Keras), scikit-learn, Kaggle, big data (Spark, Hadoop MapReduce, HDFS), matplotlib, pandas, NumPy, SciPy, Python essentials, AWS, and various command lines.

  • Updated Dec 21, 2020
  • Python
Wikipedia
Wikipedia