ai
Flax
Package description
Flax is a high-performance neural network library and ecosystem for JAX that is designed for flexibility: Try new forms of training by forking an example and by modifying the training loop, not by adding features to a framework.
Use latest version
Use specific version
PyTorch
Package description
Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch is a deep learning framework that puts Python first.
Use latest version
Use specific version
PyTorch-bundle
Package description
PyTorch with compatible versions of official Torch extensions.
Use latest version
Use specific version
Uni-Core
https://github.com/dptech-corp/Uni-Core
Package description
An efficient distributed PyTorch framework
Use latest version
Use specific version
colossalai
Package description
Colossal-AI: A Unified Deep Learning System for Big Model Era
Use latest version
Use specific version
jax
Package description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more