Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Convert Machine Learning Code Between Frameworks

Making large AI models cheaper, faster and more accessible

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

Time-Lapse Disentanglement With Conditional GANs [SIGGRAPH 2022]

Convert a PyTorch model and train it in JavaScript in your browser using ONNX Runtime Web

A WebGPU-accelerated ONNX inference run-time written 100% in Rust, ready for native and the web

Hera makes Python code easy to orchestrate on Argo Workflows through native Python integrations. It lets you construct and submit your Workflows entirely in Python. ⭐️ Remember to star!

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, 20+ clouds, or on-prem).