15 results found Sort:
- Filter by Primary Language:
- Python (6)
- Jupyter Notebook (4)
- C++ (1)
- Elixir (1)
- Go (1)
- Zig (1)
- +
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
Created
2020-10-29
2,161 commits to main branch, last one 18 days ago
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Created
2018-11-05
5,337 commits to master branch, last one 7 days ago
Any model. Any hardware. Zero compromise. Built with @ziglang / @openxla / MLIR / @bazelbuild
Created
2024-09-17
103 commits to master branch, last one 13 days ago
JAX - A curated list of resources https://github.com/google/jax
Created
2020-12-20
246 commits to main branch, last one 8 months ago
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 t...
Created
2021-10-28
33 commits to main branch, last one 2 years ago
GoMLX: An Accelerated Machine Learning Framework For Go
Created
2023-04-23
1,365 commits to main branch, last one 4 days ago
Felafax is building AI infra for non-NVIDIA GPUs
Created
2024-06-17
761 commits to main branch, last one 15 hours ago
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Created
2020-07-21
309 commits to master branch, last one 12 days ago
Extending JAX with custom C++ and CUDA code
Created
2021-01-07
56 commits to main branch, last one 4 months ago
ALBERT model Pretraining and Fine Tuning using TF2.0
Created
2019-10-31
56 commits to master branch, last one 5 years ago
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
Created
2021-08-17
97 commits to master branch, last one 2 years ago
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
Created
2020-09-25
252 commits to main branch, last one 6 days ago
基于tensorflow1.x的预训练模型调用，支持单机多卡、梯度累积，XLA加速，混合精度。可灵活训练、验证、预测。
Created
2020-09-21
94 commits to master branch, last one 3 years ago
PyTorch distributed training acceleration framework
Created
2024-07-25
41 commits to main branch, last one 7 days ago
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
Created
2023-05-28
73 commits to main branch, last one about a year ago