Here are
41 public repositories
matching this topic...
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
-
Updated
Jun 12, 2024
-
Elixir
Enabling PyTorch on XLA Devices (e.g. Google TPU)
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
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ?
-
Updated
Jun 13, 2024
-
Python
Extending JAX with custom C++ and CUDA code
-
Updated
Nov 3, 2023
-
Python
-
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
GoMLX -- Accelerated ML Libraries for Go
基于tensorflow1.x的???模型?用,支持?机多?、梯度累?,XLA加速,混合精度。可?活??、??、??。
-
Updated
Aug 16, 2021
-
Python
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
-
Updated
Jun 5, 2024
-
Jupyter Notebook
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
-
Updated
Aug 27, 2023
-
Jupyter Notebook
JAX bindings for the NVIDIA cuDecomp library
-
Updated
Jun 13, 2024
-
Python
Official scala pool repository
-
Updated
Sep 30, 2023
-
JavaScript
XLA integration of Open Neural Network Exchange (ONNX)
Tensorflow2 training code with jit compiling on multi-GPU.
-
Updated
Jan 28, 2021
-
Python
Contains materials for my talk "You don't know TensorFlow".
-
Updated
Mar 19, 2023
-
Jupyter Notebook
Provides code to serialize the different models involved in Stable Diffusion as SavedModels and to compile them with XLA.
-
Updated
Feb 27, 2023
-
Python
Improve this page
Add a description, image, and links to the
xla
topic page so that developers can more easily learn about it.
Curate this topic
Add this topic to your repo
To associate your repository with the
xla
topic, visit your repo's landing page and select "manage topics."
Learn more
You can’t perform that action at this time.