Pinned Repositories
AdaMod-tf
Tensorflow version of the AdaMod Optimizer
diffGrad-tf
DiffGrad Optimizer in Tensorflow
easydel-experiments
flat-sophia
sophia optimizer further projected towards flat areas of loss landscape
image-classification-jax
Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet
kron_torch
An implementation of PSGD Kron second-order optimizer for PyTorch
llm-jax
Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
ordax
A collection of second-order optimizers and experiments in JAX
psgd_jax
Implementation of PSGD optimizer in JAX
sophia-schedulefree-jax
Sophia optimizer with ScheduleFree
evanatyourservice's Repositories
evanatyourservice/psgd_jax
Implementation of PSGD optimizer in JAX
evanatyourservice/kron_torch
An implementation of PSGD Kron second-order optimizer for PyTorch
evanatyourservice/llm-jax
Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
evanatyourservice/sophia-schedulefree-jax
Sophia optimizer with ScheduleFree
evanatyourservice/xLSTM-JAX
An implementation of mLSTM from xLSTM in JAX
evanatyourservice/easydel-experiments
evanatyourservice/flat-sophia
sophia optimizer further projected towards flat areas of loss landscape
evanatyourservice/image-classification-jax
Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet
evanatyourservice/ordax
A collection of second-order optimizers and experiments in JAX
evanatyourservice/big_vision
Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
evanatyourservice/clip-jax
Train vision models using JAX and 🤗 transformers
evanatyourservice/EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
evanatyourservice/gpt-jax
Jax/Flax rewrite of Karpathy's nanoGPT
evanatyourservice/maxtext
A simple, performant and scalable Jax LLM!
evanatyourservice/PufferLib
Simplifying reinforcement learning for complex game environments
evanatyourservice/algorithmic-efficiency
MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.
evanatyourservice/mctx
Monte Carlo tree search in JAX
evanatyourservice/optax
Optax is a gradient processing and optimization library for JAX.
evanatyourservice/brax-trainer
evanatyourservice/datasets
🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools
evanatyourservice/dit-vs-unet
DiT vs UNet
evanatyourservice/EasyDeL
Accelerate, Optimize performance with streamlined training and serving options with JAX.
evanatyourservice/gemma
Open weights LLM from Google DeepMind.
evanatyourservice/HeavyBall
Efficient optimizers
evanatyourservice/HighPerfLLMs2024
evanatyourservice/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
evanatyourservice/levanter
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
evanatyourservice/pgx
♟️ Vectorized RL game environments in JAX
evanatyourservice/riemannian_dog
Learning-Rate-Free Stochastic Riemannian Optimization in JAX.
evanatyourservice/schedule_free
Schedule-Free Optimization in PyTorch