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 31,295 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 Jan 28, 2021
  • Python
Wikipedia
Wikipedia