Pinned Repositories
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
maxtext
A simple, performant and scalable Jax LLM!
ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
llama3
The official Meta Llama 3 GitHub site
maxtext
A simple, performant and scalable Jax LLM!
ml-auto-solutions
A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across different frameworks.
ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
xla
Enabling PyTorch on Google TPU
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
khatwanimohit's Repositories
khatwanimohit/llama3
The official Meta Llama 3 GitHub site
khatwanimohit/maxtext
A simple, performant and scalable Jax LLM!
khatwanimohit/ml-auto-solutions
A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across different frameworks.
khatwanimohit/ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
khatwanimohit/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
khatwanimohit/xla
Enabling PyTorch on Google TPU