Awesome Open Source

Programming Languages

Search results for python xla

9 search results found

Get Started With Jax ⭐ 487

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.

Extending Jax ⭐ 293

Extending JAX with custom C++ and CUDA code

Albert Tf2.0 ⭐ 155

ALBERT model Pretraining and Fine Tuning using TF2.0

Tensorflow Optimized Wheels ⭐ 116

TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA

Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload

基于tensorflow1.x的预训练模型调用，支持单机多卡、梯度累积，XLA加速，混合精度。可灵活

Jaxdecomp ⭐ 18

JAX bindings for the NVIDIA cuDecomp library

Onnx Xla ⭐ 11

XLA integration of Open Neural Network Exchange (ONNX)

Tf2 Jit Compile On Multi Gpu ⭐ 9

Tensorflow2 training code with jit compiling on multi-GPU.

Related Searches

Python Django (28,897)

Python Machine Learning (20,195)

Python Deep Learning (18,856)

Python Flask (17,643)

Python Dataset (14,792)

Python Pytorch (14,667)

Python Tensorflow (14,451)

Python Docker (14,113)

Python Command Line (13,351)

Python Jupyter Notebook (12,976)

1-9 of 9 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.