EnvironmentDescriptionQuickstart
torchEnvironment equipped with PyTorch, a powerful open-source tensor library for training neural network models written for Python.
jaxEnvironment set up for JAX, a library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.
tensorflowEnvironment designed for TensorFlow, an end-to-end open-source tensor library for deep learning in Python.
huggingfaceEnvironment optimized for Hugging Face tools like transformers, datasets, accelerate, and optimum powered by PyTorch.
fastaiEnvironment for Fast.ai tools and PyTorch, ideal for training and running inference on neural network models.