Aakash Kumar Nain's repositories
annotated_research_papers
This repo contains annotated research papers that I found really good and useful
TF_JAX_tutorials
All about the fundamental blocks of TF and JAX!
diffusion_models
All about the fundamentals and working of Diffusion Models
skin_cancer_detection
Detecting cancers on mobile phones
AugMix_TF2
TF2.0 port for Augmix paper
mistral_jax
This is a port of Mistral-7B model in JAX
CycleGAN_TF2
Implementation of CycleGAN in tf.keras
transformers
A collection of resources to study Transformers in depth.
keras-core
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
Agent-E
Automation that is driven by one or more agents
MIT000
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache-2.0000