diff --git a/precondition/__init__.py b/precondition/__init__.py index ee221b6..01c8f63 100644 --- a/precondition/__init__.py +++ b/precondition/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/distributed_shampoo.py b/precondition/distributed_shampoo.py index ac8465e..92884ab 100644 --- a/precondition/distributed_shampoo.py +++ b/precondition/distributed_shampoo.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/distributed_shampoo_test.py b/precondition/distributed_shampoo_test.py index 6781774..4196348 100644 --- a/precondition/distributed_shampoo_test.py +++ b/precondition/distributed_shampoo_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/oco/algorithms.py b/precondition/oco/algorithms.py index 7ea4378..16a5b53 100644 --- a/precondition/oco/algorithms.py +++ b/precondition/oco/algorithms.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/oco/datasets.py b/precondition/oco/datasets.py index d2e6397..20e2dfe 100644 --- a/precondition/oco/datasets.py +++ b/precondition/oco/datasets.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/oco/sweep.py b/precondition/oco/sweep.py index 59d12cf..0baa9b5 100644 --- a/precondition/oco/sweep.py +++ b/precondition/oco/sweep.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/oco/train.py b/precondition/oco/train.py index 2057ea5..34cccf8 100644 --- a/precondition/oco/train.py +++ b/precondition/oco/train.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/quantization_utils.py b/precondition/quantization_utils.py index f6075ef..d721df3 100644 --- a/precondition/quantization_utils.py +++ b/precondition/quantization_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/sm3.py b/precondition/sm3.py index 414fe10..9387af8 100644 --- a/precondition/sm3.py +++ b/precondition/sm3.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/sm3_test.py b/precondition/sm3_test.py index 2fde41f..95814cd 100644 --- a/precondition/sm3_test.py +++ b/precondition/sm3_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/grafting.py b/precondition/tearfree/grafting.py index dedf2f9..a607c89 100644 --- a/precondition/tearfree/grafting.py +++ b/precondition/tearfree/grafting.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/grafting_test.py b/precondition/tearfree/grafting_test.py index 51375cc..69cbeb3 100644 --- a/precondition/tearfree/grafting_test.py +++ b/precondition/tearfree/grafting_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/momentum.py b/precondition/tearfree/momentum.py index 234229f..93a241d 100644 --- a/precondition/tearfree/momentum.py +++ b/precondition/tearfree/momentum.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/momentum_test.py b/precondition/tearfree/momentum_test.py index b16db5d..29a7fcf 100644 --- a/precondition/tearfree/momentum_test.py +++ b/precondition/tearfree/momentum_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/optimizer.py b/precondition/tearfree/optimizer.py index fb1954e..759059f 100644 --- a/precondition/tearfree/optimizer.py +++ b/precondition/tearfree/optimizer.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/optimizer_smoke_test.py b/precondition/tearfree/optimizer_smoke_test.py index 6e6543e..a259986 100644 --- a/precondition/tearfree/optimizer_smoke_test.py +++ b/precondition/tearfree/optimizer_smoke_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/optimizer_test.py b/precondition/tearfree/optimizer_test.py index cbcc166..2d3ed2d 100644 --- a/precondition/tearfree/optimizer_test.py +++ b/precondition/tearfree/optimizer_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/praxis_shim.py b/precondition/tearfree/praxis_shim.py index 3c15295..3e7394d 100644 --- a/precondition/tearfree/praxis_shim.py +++ b/precondition/tearfree/praxis_shim.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/reallocation.py b/precondition/tearfree/reallocation.py index 591d083..355faed 100644 --- a/precondition/tearfree/reallocation.py +++ b/precondition/tearfree/reallocation.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/reallocation_test.py b/precondition/tearfree/reallocation_test.py index 6354d64..7d87f2a 100644 --- a/precondition/tearfree/reallocation_test.py +++ b/precondition/tearfree/reallocation_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/reshaper.py b/precondition/tearfree/reshaper.py index 551ea2e..db8bc8e 100644 --- a/precondition/tearfree/reshaper.py +++ b/precondition/tearfree/reshaper.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/reshaper_test.py b/precondition/tearfree/reshaper_test.py index dca075e..94d27ad 100644 --- a/precondition/tearfree/reshaper_test.py +++ b/precondition/tearfree/reshaper_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/second_order.py b/precondition/tearfree/second_order.py index 027333f..63f1e2f 100644 --- a/precondition/tearfree/second_order.py +++ b/precondition/tearfree/second_order.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/shampoo.py b/precondition/tearfree/shampoo.py index 6bc65a3..f5fff14 100644 --- a/precondition/tearfree/shampoo.py +++ b/precondition/tearfree/shampoo.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/shampoo_test.py b/precondition/tearfree/shampoo_test.py index ca46678..db81222 100644 --- a/precondition/tearfree/shampoo_test.py +++ b/precondition/tearfree/shampoo_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -289,8 +289,6 @@ def test_precond_freq(self): rng1, rng2 = jax.random.split(key, 2) seq1 = jnp.arange(freq, dtype=int) seq2 = jnp.copy(seq1) - jax.random.shuffle(rng1, seq1) - jax.random.shuffle(rng2, seq2) # Shuffle within groups of shuffled = jnp.take(grads, jnp.concatenate([seq1, seq2 + freq]), axis=0) diff --git a/precondition/tearfree/sketchy.py b/precondition/tearfree/sketchy.py index 48a2c79..37bf1bb 100644 --- a/precondition/tearfree/sketchy.py +++ b/precondition/tearfree/sketchy.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/precondition/tearfree/sketchy_test.py b/precondition/tearfree/sketchy_test.py index 75589fe..3a71835 100644 --- a/precondition/tearfree/sketchy_test.py +++ b/precondition/tearfree/sketchy_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The precondition Authors. +# Copyright 2024 The precondition Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.