xla

TPU accelerator

Enables PyTorch deep learning framework on XLA devices (e.g., Google TPU) for efficient and scalable training

Enabling PyTorch on XLA Devices (e.g. Google TPU)

GitHub

2k stars
58 watching
480 forks
Language: C++
last commit: 6 days ago
compilerdeep-learningpytorchxla

Related projects:

Repository Description Stars
openxla/xla Compiles machine learning models for high-performance execution on various hardware platforms 2,699
lazziya/xlocalizer.samples Demonstrates localizing Asp.Net Core web applications using XLocalizer 13
huggingface/pytorch-openai-transformer-lm Implementing OpenAI's transformer language model in PyTorch with pre-trained weights and fine-tuning capabilities 1,511
openai/finetune-transformer-lm This project provides code and model for improving language understanding through generative pre-training using a transformer-based architecture. 2,160
clementfarabet/lua---nnx An extension to Torch7's neural network package with experimental modules and optimizations. 97
iglaweb/tfprofiler An app for profiling and optimizing the performance of TensorFlow Lite models on mobile devices 27
plentz/lol_dba A tool that analyzes application models to identify columns that likely need indexing and generates migration scripts for adding or removing these indexes. 1,585
neopallium/lua-pb A Lua implementation of Protocol Buffers for efficient data serialization and deserialization. 289
chrisa/lua-usdt Bindings to interact with the USDT (Unified Sampling and Tracing Daemon) tracing system from Lua. 6
deltaf1/uxn-lua Port of Uxn Virtual Machine to Lua and Love2D game engine 15
lge-arc-advancedai/auptimizer Automates model building and deployment process by optimizing hyperparameters and compressing models for edge computing. 200
locuslab/trellisnet This repository presents a novel neural network architecture and its applications in sequence modeling tasks such as language modeling and classification. 473
huggingface/tflite-android-transformers Converts popular transformer models to run on Android devices for efficient inference and generation tasks. 392
alexpasmantier/pymple.nvim Automatically updates imports and resolves missing imports in Python files when the file or directory is moved or renamed 76
exo-explore/exo Allows developers to run AI models on personal devices with diverse hardware configurations. 14,829