steventk-g's repositories
actnn
ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
Language:PythonMIT000
llama
Inference code for LLaMA models
Language:PythonGPL-3.0000
ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
Language:JsonnetApache-2.0000
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++NOASSERTION000
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Language:PythonApache-2.0000
trlx
A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)
Language:PythonMIT000
xla
Enabling PyTorch on Google TPU
Language:C++NOASSERTION000