TensorFlow

Tensorflow is an end-to-end open source platform for machine learning using CPUs and GPUS.

PyTorch

PyTorch is an optimized tensor library for deep learning using CPUs and GPUs.

Hugging Face

Hugging Face is a collaboration platform for the AI community. It helps users build, train, and deploy machine learning models. The platform also serves as a repository for sharing datasets and models. Users are able to interface with Hugging Face datasets and models with both TensorFlow and PyTorch.

JAX

JAX is a Python library designed for high-performance machine learning research on CPUs and GPUs. It offers a lightweight NumPy-like API for array based computing with composable function transformations.

Academic Machine Learning Environment

We have created a conda environment that contains popular machine learning libraries (including all of the above mentioned packages). The packages in this environment are updated each semester.