JaxLab

ML from scratch implementations in the awesome library, JAX!

10 ML Algorithms to Implement in JAX

• Linear Regression
• Logistic Regression
• Softmax Regression (Multiclass Classification)
• K-Nearest Neighbors (KNN)
• Support Vector Machine (SVM)
• Principal Component Analysis (PCA)
• K-Means Clustering
• Naive Bayes Classifier
• Decision Trees