ML from scratch implementations in the awesome library, 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