keras-team / tf-keras
The TensorFlow-specific implementation of the Keras API, which was the default Keras from 2019 to 2023.
☆64Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for tf-keras
- ☆77Updated 8 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated 2 months ago
- This repository contains example code to build models on TPUs☆30Updated last year
- Plugin for deploying MLflow models to TorchServe☆106Updated last year
- PyTorch RFCs (experimental)☆130Updated 2 months ago
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- ☆56Updated 2 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆215Updated this week
- FIL backend for the Triton Inference Server☆72Updated this week
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆83Updated this week
- Documentation autogeneration utilities.☆30Updated 2 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆216Updated last week
- ☆178Updated last week
- The Triton backend for the ONNX Runtime.☆132Updated this week
- JAX-Toolbox☆245Updated this week
- The Triton backend for the PyTorch TorchScript models.☆127Updated this week
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆34Updated 2 years ago
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64Updated 2 months ago
- ☆64Updated 2 years ago
- Productionize machine learning predictions, with ONNX or without☆66Updated 10 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated 2 weeks ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆175Updated last week
- Implementation of Flash Attention in Jax☆196Updated 8 months ago
- Torch Distributed Experimental☆116Updated 3 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆270Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆303Updated this week
- Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.☆36Updated last year
- A profiling and performance analysis tool for TensorFlow☆360Updated this week
- Accelerate PyTorch models with ONNX Runtime☆356Updated 2 months ago