xla
Here are 47 public repositories matching this topic...
Enabling PyTorch on XLA Devices (e.g. Google TPU)
-
Updated
Dec 5, 2024 - C++
JAX - A curated list of resources https://github.com/google/jax
-
Updated
Jul 10, 2024
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
-
Updated
Nov 29, 2023 - Jupyter Notebook
GoMLX: An Accelerated Machine Learning Framework For Go
-
Updated
Dec 4, 2024 - Go
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
-
Updated
Nov 4, 2024 - Python
Julia on TPUs
-
Updated
Dec 18, 2020 - Julia
ALBERT model Pretraining and Fine Tuning using TF2.0
-
Updated
Mar 24, 2023 - Python
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
-
Updated
Aug 6, 2022 - Python
TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA
-
Updated
Feb 12, 2020
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
-
Updated
Nov 15, 2024 - Jupyter Notebook
基于tensorflow1.x的预训练模型调用,支持单机多卡、梯度累积,XLA加速,混合精度。可灵活训练、验证、预测。
-
Updated
Aug 16, 2021 - Python
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
-
Updated
Aug 27, 2023 - Jupyter Notebook
PyTorch distributed training acceleration framework
-
Updated
Dec 5, 2024 - Python
Official scala pool repository
-
Updated
Oct 4, 2024 - JavaScript
XLA integration of Open Neural Network Exchange (ONNX)
-
Updated
Aug 17, 2018 - C++
Improve this page
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."