Pinned Repositories
bi-tempered-loss
Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf
init2winit
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
orbax
Orbax provides common checkpointing and persistence utilities for JAX users
praxis
trax
Trax — Deep Learning with Clear Code and Speed
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
keras-contrib
Keras community contributions
rohan-anil's Repositories
rohan-anil/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
rohan-anil/keras-contrib
Keras community contributions