-
Debugging a Machine Learning Architecture - A Step-by-Step Guide
a short guide on making sure everything works out for your ML solution
-
Llama-2 finetuning using PEFT and Multi-GPU setup
repository for the implementation of PEFT finetuning using jax
-
training large language models using distributed resources
With the increasing popularity of large language models, it becomes increasingly important to use all available resources to gain an edge in training.
-
a post with giscus comments
an example of a blog post with giscus comments
-
understanding jax's scan function
jax.lax.scan is a function which allows jit-able loops