Skip to content

Commit 304b309

Browse files
authored
Create README.md
1 parent 92fa955 commit 304b309

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# Jax-logistic-regression
2+
Logistic regression classifier using JAX to support GPU acceleration.
3+
4+
This class is an update of a logistic regression class used in my intro to machine learning course. The major difference is the handling of the gradient descent operations,
5+
which were rewritten using jax's grad, jit, and vmap functions. The goal with this project is speed - I've found that using JaxReg with GPU acceleration gives a ~15x speed
6+
increase over the original class.

0 commit comments

Comments
 (0)