evanatyourservice | GitHub owners

evanatyourservice/quad_torch

An implementation of PSGD-QUAD optimizer for PyTorch

Language: Python - Size: 16.6 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 4 - Forks: 0

evanatyourservice/clip-jax Fork of borisdayma/clip-jax

Train vision models using JAX and 🤗 transformers

Language: Python - Size: 1.01 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 1 - Forks: 0

evanatyourservice/kron_torch

An implementation of PSGD Kron second-order optimizer for PyTorch

Language: Python - Size: 303 KB - Last synced at: 2 days ago - Pushed at: about 1 month ago - Stars: 96 - Forks: 6

evanatyourservice/gidd Fork of dvruette/gidd

Code accompanying the paper "Generalized Interpolating Discrete Diffusion"

Language: Python - Size: 627 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 1 - Forks: 0

evanatyourservice/gidd-easydel Fork of dvruette/gidd-easydel

Language: Python - Size: 191 KB - Last synced at: 28 days ago - Pushed at: 28 days ago - Stars: 0 - Forks: 0

evanatyourservice/levanter Fork of stanford-crfm/levanter

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

Language: Python - Size: 8.79 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 1 - Forks: 0

evanatyourservice/psgd_jax

Implementation of PSGD optimizer in JAX

Language: Python - Size: 329 KB - Last synced at: 7 days ago - Pushed at: 8 months ago - Stars: 34 - Forks: 2

evanatyourservice/torch-grokking Fork of stockeh/mlx-grokking

Grokking on modular arithmetic in less than 150 epochs in MLX

Language: Python - Size: 1.69 MB - Last synced at: 2 months ago - Pushed at: 2 months ago - Stars: 0 - Forks: 0

evanatyourservice/tiny-muon-jax

Language: Python - Size: 2.93 KB - Last synced at: 3 months ago - Pushed at: 5 months ago - Stars: 2 - Forks: 0

evanatyourservice/image-classification-jax

Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet

Language: Python - Size: 64.5 KB - Last synced at: 12 days ago - Pushed at: 8 months ago - Stars: 3 - Forks: 0

evanatyourservice/jax-torch-comparison

Language: Python - Size: 382 KB - Last synced at: 5 months ago - Pushed at: 7 months ago - Stars: 6 - Forks: 3

evanatyourservice/PufferLib Fork of PufferAI/PufferLib

Simplifying reinforcement learning for complex game environments

Language: C - Size: 169 MB - Last synced at: 5 months ago - Pushed at: 5 months ago - Stars: 0 - Forks: 0

evanatyourservice/algorithmic-efficiency Fork of mlcommons/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.

Language: Python - Size: 29.4 MB - Last synced at: 6 months ago - Pushed at: 6 months ago - Stars: 0 - Forks: 0

evanatyourservice/distributed_kron

An implementation of PSGD Kron in JAX for distributed training in JAX or Flax

Language: Python - Size: 2.29 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 9 - Forks: 0

evanatyourservice/beta-tcvae Fork of rtqichen/beta-tcvae

code for "Isolating Sources of Disentanglement in Variational Autoencoders".

Language: Python - Size: 24.4 KB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 0 - Forks: 0

evanatyourservice/dreamerv3 Fork of danijar/dreamerv3

Mastering Diverse Domains through World Models

Language: Python - Size: 25.5 MB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 0 - Forks: 0

evanatyourservice/modded-nanogpt Fork of KellerJordan/modded-nanogpt

NanoGPT (124M) in 3.4 minutes

Language: Python - Size: 14.2 MB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 2 - Forks: 0

evanatyourservice/llm-jax

Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.

Language: Python - Size: 3.57 MB - Last synced at: 5 months ago - Pushed at: 6 months ago - Stars: 17 - Forks: 1

evanatyourservice/PSGD_Nuon Fork of opooladz/PSGD_Nuon

Not Muon

Language: Python - Size: 0 Bytes - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 0 - Forks: 0

evanatyourservice/ppo-atari-notebook

tiny cleanrl experiment using PPO for atari to compare adam optimizer with psgd kron optimizer

Language: Jupyter Notebook - Size: 5.86 KB - Last synced at: 5 months ago - Pushed at: 7 months ago - Stars: 0 - Forks: 1

evanatyourservice/jax-flow Fork of kvfrans/jax-flow

Flow-matching algorithms in JAX

Language: Python - Size: 1.04 MB - Last synced at: 8 months ago - Pushed at: 8 months ago - Stars: 0 - Forks: 0

evanatyourservice/I-xTx-jax

playing around with minimizing the toy fn I - x ^ T x

Language: Jupyter Notebook - Size: 887 KB - Last synced at: about 1 month ago - Pushed at: 8 months ago - Stars: 1 - Forks: 0

evanatyourservice/flat-sophia

sophia optimizer further projected towards flat areas of loss landscape

Language: Python - Size: 269 KB - Last synced at: 5 months ago - Pushed at: 9 months ago - Stars: 2 - Forks: 0

evanatyourservice/xLSTM-JAX

An implementation of mLSTM from xLSTM in JAX

Language: Python - Size: 8.79 KB - Last synced at: 3 months ago - Pushed at: 9 months ago - Stars: 3 - Forks: 0

evanatyourservice/HeavyBall Fork of ClashLuke/HeavyBall

Efficient optimizers

Language: Python - Size: 1.58 MB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 0 - Forks: 0

evanatyourservice/mctx Fork of google-deepmind/mctx

Monte Carlo tree search in JAX

Language: Python - Size: 479 KB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 0 - Forks: 0

evanatyourservice/pgx Fork of sotetsuk/pgx

♟️ Vectorized RL game environments in JAX

Language: Python - Size: 74.5 MB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 0 - Forks: 0

evanatyourservice/dit-vs-unet Fork of SonicCodes/dit-vs-unet

DiT vs UNet

Language: Python - Size: 31.5 MB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 0 - Forks: 0

evanatyourservice/gpt-jax Fork of jenkspt/gpt-jax

Jax/Flax rewrite of Karpathy's nanoGPT

Language: Python - Size: 3.21 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 1 - Forks: 0

evanatyourservice/EasyLM Fork of young-geng/EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

Language: Python - Size: 529 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 1 - Forks: 0

evanatyourservice/sophia-schedulefree-jax

Sophia optimizer with ScheduleFree

Language: Python - Size: 12.7 KB - Last synced at: 5 months ago - Pushed at: about 1 year ago - Stars: 4 - Forks: 1

evanatyourservice/google-research Fork of google-research/google-research

Google Research

Language: Jupyter Notebook - Size: 598 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 0 - Forks: 0

evanatyourservice/evanatyourservice.github.io

Size: 0 Bytes - Last synced at: 6 months ago - Pushed at: almost 2 years ago - Stars: 0 - Forks: 0

evanatyourservice/ONI Fork of huangleiBuaa/ONI

PyTorch and Torch implementation for our accepted CVPR 2020 paper (Oral): Controllable Orthogonalization in Training DNNs

Size: 645 KB - Last synced at: over 1 year ago - Pushed at: over 4 years ago - Stars: 0 - Forks: 0

GitHub / evanatyourservice 1 Repository